Repository: spark Updated Branches: refs/heads/master ac84fb64d -> ccc6dc0f4
[MINOR][ML] Rename TreeEnsembleModels to TreeEnsembleModel for PySpark ## What changes were proposed in this pull request? Fix the typo of ```TreeEnsembleModels``` for PySpark, it should ```TreeEnsembleModel``` which will be consistent with Scala. What's more, it represents a tree ensemble model, so ```TreeEnsembleModel``` should be more reasonable. This should not be used public, so it will not involve breaking change. ## How was this patch tested? No new tests, should pass existing ones. Author: Yanbo Liang <yblia...@gmail.com> Closes #14454 from yanboliang/TreeEnsembleModel. Project: http://git-wip-us.apache.org/repos/asf/spark/repo Commit: http://git-wip-us.apache.org/repos/asf/spark/commit/ccc6dc0f Tree: http://git-wip-us.apache.org/repos/asf/spark/tree/ccc6dc0f Diff: http://git-wip-us.apache.org/repos/asf/spark/diff/ccc6dc0f Branch: refs/heads/master Commit: ccc6dc0f4b62837c73fca0e3c8b9c14be798b062 Parents: ac84fb6 Author: Yanbo Liang <yblia...@gmail.com> Authored: Thu Aug 11 22:39:19 2016 -0700 Committer: Yanbo Liang <yblia...@gmail.com> Committed: Thu Aug 11 22:39:19 2016 -0700 ---------------------------------------------------------------------- python/pyspark/ml/classification.py | 6 +++--- python/pyspark/ml/regression.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/spark/blob/ccc6dc0f/python/pyspark/ml/classification.py ---------------------------------------------------------------------- diff --git a/python/pyspark/ml/classification.py b/python/pyspark/ml/classification.py index 9a3c7b1..6468007 100644 --- a/python/pyspark/ml/classification.py +++ b/python/pyspark/ml/classification.py @@ -22,7 +22,7 @@ from pyspark import since, keyword_only from pyspark.ml import Estimator, Model from pyspark.ml.param.shared import * from pyspark.ml.regression import DecisionTreeModel, DecisionTreeRegressionModel, \ - RandomForestParams, TreeEnsembleModels, TreeEnsembleParams + RandomForestParams, TreeEnsembleModel, TreeEnsembleParams from pyspark.ml.util import * from pyspark.ml.wrapper import JavaEstimator, JavaModel, JavaParams from pyspark.ml.wrapper import JavaWrapper @@ -722,7 +722,7 @@ class RandomForestClassifier(JavaEstimator, HasFeaturesCol, HasLabelCol, HasPred return RandomForestClassificationModel(java_model) -class RandomForestClassificationModel(TreeEnsembleModels, JavaMLWritable, JavaMLReadable): +class RandomForestClassificationModel(TreeEnsembleModel, JavaMLWritable, JavaMLReadable): """ Model fitted by RandomForestClassifier. @@ -873,7 +873,7 @@ class GBTClassifier(JavaEstimator, HasFeaturesCol, HasLabelCol, HasPredictionCol return self.getOrDefault(self.lossType) -class GBTClassificationModel(TreeEnsembleModels, JavaMLWritable, JavaMLReadable): +class GBTClassificationModel(TreeEnsembleModel, JavaMLWritable, JavaMLReadable): """ Model fitted by GBTClassifier. http://git-wip-us.apache.org/repos/asf/spark/blob/ccc6dc0f/python/pyspark/ml/regression.py ---------------------------------------------------------------------- diff --git a/python/pyspark/ml/regression.py b/python/pyspark/ml/regression.py index d88dc75..1ae2bd4 100644 --- a/python/pyspark/ml/regression.py +++ b/python/pyspark/ml/regression.py @@ -749,7 +749,7 @@ class DecisionTreeModel(JavaModel): @inherit_doc -class TreeEnsembleModels(JavaModel): +class TreeEnsembleModel(JavaModel): """ (private abstraction) @@ -909,7 +909,7 @@ class RandomForestRegressor(JavaEstimator, HasFeaturesCol, HasLabelCol, HasPredi return RandomForestRegressionModel(java_model) -class RandomForestRegressionModel(TreeEnsembleModels, JavaMLWritable, JavaMLReadable): +class RandomForestRegressionModel(TreeEnsembleModel, JavaMLWritable, JavaMLReadable): """ Model fitted by :class:`RandomForestRegressor`. @@ -1047,7 +1047,7 @@ class GBTRegressor(JavaEstimator, HasFeaturesCol, HasLabelCol, HasPredictionCol, return self.getOrDefault(self.lossType) -class GBTRegressionModel(TreeEnsembleModels, JavaMLWritable, JavaMLReadable): +class GBTRegressionModel(TreeEnsembleModel, JavaMLWritable, JavaMLReadable): """ Model fitted by :class:`GBTRegressor`. --------------------------------------------------------------------- To unsubscribe, e-mail: commits-unsubscr...@spark.apache.org For additional commands, e-mail: commits-h...@spark.apache.org