[ https://issues.apache.org/jira/browse/SPARK-16079?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel ]
Xiangrui Meng updated SPARK-16079: ---------------------------------- Affects Version/s: 2.0.0 > PySpark ML classification missing import of DecisionTreeRegressionModel for > GBTClassificationModel > -------------------------------------------------------------------------------------------------- > > Key: SPARK-16079 > URL: https://issues.apache.org/jira/browse/SPARK-16079 > Project: Spark > Issue Type: Bug > Components: ML, PySpark > Affects Versions: 2.0.0 > Reporter: Bryan Cutler > > In GBTClassificationModel, the overloaded method {{trees}} casts the > DecisionTree to a DecisionTreeRegressionModel, however, the import for this > class is missing and leads to a {{NameError}} > {noformat} > spark/python/pyspark/ml/classification.pyc in trees(self) > 888 def trees(self): > 889 """Trees in this ensemble. Warning: These have null parent > Estimators.""" > --> 890 return [DecisionTreeRegressionModel(m) for m in > list(self._call_java("trees"))] > 891 > 892 > NameError: global name 'DecisionTreeRegressionModel' is not defined > {noformat} -- This message was sent by Atlassian JIRA (v6.3.4#6332) --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org For additional commands, e-mail: issues-h...@spark.apache.org