gpt4 book ai didi

java - ml.DecisionTreeClassificationModel 中的 mllib.DecisionTreeModel.toDebugString() 等效项

转载 作者:行者123 更新时间:2023-12-01 09:49:42 24 4
gpt4 key购买 nike

正如问题所说,是否有 Spark org.apache.spark.mllib.tree.model.DecisionTreeClassificationModel.toDebugString() 的等价物在org.apache.spark.ml.classification.DecisionTreeClassificationModel

我浏览了后者的API文档,发现了这个方法rootNode()返回 org.apache.spark.ml.tree.Node object 这似乎是一个递归对象,所以我应该使用这个类来自己构建树结构吗?

感谢期待。

最佳答案

org.apache.spark.ml.classification.DecisionTreeClassificationModel 已经实现了 toDebugString() 方法,因为它具有 DecisionTreeModel 作为特征。

示例:

class org.apache.spark.ml.classification.DecisionTreeClassificationModel
DecisionTreeClassificationModel of depth 1 with 3 nodes
If (feature 378 <= 71.0)
Predict: 1.0
Else (feature 378 > 71.0)
Predict: 0.0

关于java - ml.DecisionTreeClassificationModel 中的 mllib.DecisionTreeModel.toDebugString() 等效项,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37678365/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com