Bryan Cutler created SPARK-16079:
------------------------------------
Summary: 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
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
