显示在 pyspark ml 库中建模的决策树规则

Question

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

谢谢。

Answer 1

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