From 540fe8e2569b8ae405be5c264b1df1f883eac7a3 Mon Sep 17 00:00:00 2001 From: GayathriMurali Date: Wed, 23 Mar 2016 13:34:35 -0700 Subject: [PATCH] Merged to master. Added persistence to DecisionTreeRegressionModel --- python/pyspark/ml/regression.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/pyspark/ml/regression.py b/python/pyspark/ml/regression.py index 891ebba936bdf..be5827e079b5f 100644 --- 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.