[jira] [Updated] (SPARK-16079) PySpark ML classification missing import of DecisionTreeRegressionModel for GBTClassificationModel

["Xiangrui Meng (JIRA)" <ji...@apache.org>]

     [ 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-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org