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

Question

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

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

非常感谢。

Answer 1

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

Param<String> leafCol()

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