Spark MLlib 中的决策树可以返回每个样本预测为的叶索引吗？

【问题标题】：Can decision tree in Spark MLlib return leaf index that each sample is predicted as?Spark MLlib 中的决策树可以返回每个样本预测为的叶索引吗？
【发布时间】：2018-09-22 08:55:20
【问题描述】：

我在带有 Scala 的 Spark MLlib 中使用决策树。有没有实现的方法返回每个样本被预测为的叶子的索引？

我知道在 Scikit-learn 中，方法 {apply} 能够做到这一点。我想知道 MLlib 中是否也存在类似的方法。

非常感谢。

【问题讨论】：

标签： apache-spark apache-spark-mllib decision-tree

【解决方案1】：

您可以设置模型将使用的“leafCol”参数来填充每个样本的叶子 id。 documentation 说：

Param<String> leafCol()

Leaf indices column name. Predicted leaf index of each instance in each tree by preorder. (default = "")

【讨论】：