显示在pyspark ml库

Question

我是新来的火花。我使用基于Dataframe的API（即pyspark.ml）建模了决策树。我希望显示决策树的规则，类似于我们使用 toDebugString 在spark中基于RDD的API（spark.mllib）获得的规则。 我已阅读文档，但无法找到如何显示规则。还有其他办法吗？

谢谢。

Answer 1

从Spark 2.0开始，DecisionTreeClassificationModel和DecisionTreeRegressionModel都提供toDebugString方法。