[
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
