用 List<Double> 预测

Question

我有一个来自 libsvm 文件的训练有素的 RandomForestModel 对象，现在想使用该模型调用 predict 方法，并使用 List<Double> 作为特征。如何将List<Double> 转换为 Java 中的必要输入并查看预测的目标类？

据我所知，predict(...) 接受JavaRDD<Vector>，但我不确定如何将List<Double> 转换为那个。

Answer 1

通过浏览docs，看起来RandomForestModel 也可以在Vector 上调用predict。您可以将List<Double> 转换为DenseVector (docs)，方法是将列表转换为double 数组，如this 问题，然后执行new DenseVector(double_array)。

Answer 2

根据数据的稀疏性，您可以使用 SparseVector 或 DenseVector 编写从 List 到 Vector 的 map 转换

int length = yourList.size();
double[] inputArray = new double[length];
yourList.toArray(inputArray);
Vector inputVector = new DenseVector(inputArray);