在Spark（Scala）中查看决策树，就像-then-else字符串

Question

我使用库DecisionTreeRegressor在pyspark（dtModel）中构建了一个决策树，可以使用代码以if-then-else格式查看

print dtModel.stages[-1]._java_obj.toDebugString()

如果我使用相同的库（DecisionTreeRegressor），如何使用scala查看相同内容。

使用代码

println(dtModel.stages(-1).asInstanceOf[DecisionTreeRegressor].toDebugString())

给出错误

<console>:62: error: value stages is not a member of org.apache.spark.ml.Model[_$4]
              println(dtModel.stages(-1).asInstanceOf[DecisionTreeRegressor].toDebugString())
                              ^

Answer 1

使用以下命令：

dtModel.stages(2).asInstanceOf[org.apache.spark.ml.regression.DecisionTreeRegressionModel].toDebugString