The Apache Spark's documentation(1.4.0)承诺随机森林(决策树的承诺相同)可以扩展到多类分类设置。但是,我找不到使用RandomForestModel API达到此目的的任何方法:
randomForestModel.predict
如果在1.4.0版本中真的有可能,请你提供一个正确的方法吗?
答案 0 :(得分:0)
好的,我在实验包(ml)中找到了它:org.apache.spark.ml.classification.RandomForestClassificationModel
应该使用和transform来代替预测:
RandomForestClassificationModel.transform