私はlibsvmファイルから訓練されたRandomForestModel
オブジェクトを持っており、List<Double>
を機能として使用して、そのモデルでpredictメソッドを呼び出すことになります。 List<Double>
をJavaの必要な入力に変換し、予測されたターゲットクラスを表示するにはどうすればよいですか?リストを使って予測する<Double>
予測できるものは... JavaRDD<Vector>
ですが、List<Double>
に変換する方法は不明です。
私はlibsvmファイルから訓練されたRandomForestModel
オブジェクトを持っており、List<Double>
を機能として使用して、そのモデルでpredictメソッドを呼び出すことになります。 List<Double>
をJavaの必要な入力に変換し、予測されたターゲットクラスを表示するにはどうすればよいですか?リストを使って予測する<Double>
予測できるものは... JavaRDD<Vector>
ですが、List<Double>
に変換する方法は不明です。
はあなたのデータのスパース性に依存し、あなたはSparseVectorまたはDenseVector
int length = yourList.size();
double[] inputArray = new double[length];
yourList.toArray(inputArray);
Vector inputVector = new DenseVector(inputArray);
のいずれかを使用してリストからベクトルへの変換
map
を書くことができ、私は私が使うべきものがhttpsであると仮定しています://spark.apache .org/docs/2.2.0/api/java/org/apache/spark/mllib/tree/model/RandomForestModel.html#predict-org.apache.spark.api.java.JavaRDD- – rimskyドキュメントを読むことをお勧めしますJavaRDDクラスの –