Merged to master. Added persistence to DecisionTreeRegressionModel

apache · Mar 23, 2016 · 540fe8e · 540fe8e
1 parent 0154444
commit 540fe8e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/pyspark/ml/regression.py b/python/pyspark/ml/regression.py
@@ -466,7 +466,7 @@ def _create_model(self, java_model):
 
 
 @inherit_doc
-class DecisionTreeModel(JavaModel, MLWritable, MLReadable):
+class DecisionTreeModel(JavaModel):
     """Abstraction for Decision Tree models.
 
     .. versionadded:: 1.5.0
@@ -506,7 +506,7 @@ def __repr__(self):
 
 
 @inherit_doc
-class DecisionTreeRegressionModel(DecisionTreeModel):
+class DecisionTreeRegressionModel(DecisionTreeModel, MLWritable, MLReadable):
     """
     Model fitted by DecisionTreeRegressor.