Просмотреть дерево принятия решений в Spark (Scala), как если бы строка-then-else

Question

Я построил дерево решений в pyspark (dtModel), используя библиотеку DecisionTreeRegressor, и его можно просмотреть в формате if-then-else, используя код

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

Как я могу просматривать то же самое с помощью scala, если я использую одну и ту же библиотеку (DecisionTreeRegressor).

Использование кода

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