【问题标题】:Equivalent of mllib.DecisionTreeModel.toDebugString() in ml.DecisionTreeClassificationModel等效于 ml.DecisionTreeClassificationModel 中的 mllib.DecisionTreeModel.toDebugString()
【发布时间】:2016-10-07 07:07:49
【问题描述】:

正如问题所说,org.apache.spark.ml.classification.DecisionTreeClassificationModel 中是否有任何等效的 Spark org.apache.spark.mllib.tree.model.DecisionTreeClassificationModel.toDebugString()

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

感谢期待。

【问题讨论】:

    标签: java apache-spark decision-tree apache-spark-mllib


    【解决方案1】:

    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 API文档中没有提到这个方法...
    猜你喜欢
    • 2019-01-24
    • 1970-01-01
    • 2021-05-26
    • 2017-09-07
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多