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

["Bryan Cutler (JIRA)" <ji...@apache.org>]

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