Apache Spark中随机森林的多类分类

时间:2015-07-06 10:54:04

标签: apache-spark random-forest decision-tree apache-spark-mllib multilabel-classification

The Apache Spark's documentation(1.4.0)承诺随机森林(决策树的承诺相同)可以扩展到多类分类设置。但是,我找不到使用RandomForestModel API达到此目的的任何方法:

randomForestModel.predict

如果在1.4.0版本中真的有可能,请你提供一个正确的方法吗?

1 个答案:

答案 0 :(得分:0)

好的,我在实验包(ml)中找到了它:org.apache.spark.ml.classification.RandomForestClassificationModel

应该使用

和transform来代替预测:

RandomForestClassificationModel.transform