diff options
author | Xiangrui Meng <meng@databricks.com> | 2015-10-06 14:58:42 -0700 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2015-10-06 14:58:42 -0700 |
commit | 5e035403d41527f092705c68f0dd1b4b946014d6 (patch) | |
tree | 90cd8effa3230222e2a4ffa9d3fe3c44654005ec /python/pyspark | |
parent | 5952bdb7df20d007d59f82261095faca3822c6f6 (diff) | |
download | spark-5e035403d41527f092705c68f0dd1b4b946014d6.tar.gz spark-5e035403d41527f092705c68f0dd1b4b946014d6.tar.bz2 spark-5e035403d41527f092705c68f0dd1b4b946014d6.zip |
[SPARK-10957] [ML] setParams changes quantileProbabilities unexpectly in PySpark's AFTSurvivalRegression
If user doesn't specify `quantileProbs` in `setParams`, it will get reset to the default value. We don't need special handling here. vectorijk yanboliang
Author: Xiangrui Meng <meng@databricks.com>
Closes #9001 from mengxr/SPARK-10957.
Diffstat (limited to 'python/pyspark')
-rw-r--r-- | python/pyspark/ml/regression.py | 6 |
1 files changed, 1 insertions, 5 deletions
diff --git a/python/pyspark/ml/regression.py b/python/pyspark/ml/regression.py index a0f7f54e65..e12abeba01 100644 --- a/python/pyspark/ml/regression.py +++ b/python/pyspark/ml/regression.py @@ -701,11 +701,7 @@ class AFTSurvivalRegression(JavaEstimator, HasFeaturesCol, HasLabelCol, HasPredi quantilesCol=None): """ kwargs = self.setParams._input_kwargs - if quantileProbabilities is None: - return self._set(**kwargs).setQuantileProbabilities([0.01, 0.05, 0.1, 0.25, 0.5, - 0.75, 0.9, 0.95, 0.99]) - else: - return self._set(**kwargs) + return self._set(**kwargs) def _create_model(self, java_model): return AFTSurvivalRegressionModel(java_model) |