Repository: spark Updated Branches: refs/heads/master 5952bdb7d -> 5e035403d
[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 <[email protected]> Closes #9001 from mengxr/SPARK-10957. Project: http://git-wip-us.apache.org/repos/asf/spark/repo Commit: http://git-wip-us.apache.org/repos/asf/spark/commit/5e035403 Tree: http://git-wip-us.apache.org/repos/asf/spark/tree/5e035403 Diff: http://git-wip-us.apache.org/repos/asf/spark/diff/5e035403 Branch: refs/heads/master Commit: 5e035403d41527f092705c68f0dd1b4b946014d6 Parents: 5952bdb Author: Xiangrui Meng <[email protected]> Authored: Tue Oct 6 14:58:42 2015 -0700 Committer: Xiangrui Meng <[email protected]> Committed: Tue Oct 6 14:58:42 2015 -0700 ---------------------------------------------------------------------- python/pyspark/ml/regression.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/spark/blob/5e035403/python/pyspark/ml/regression.py ---------------------------------------------------------------------- diff --git a/python/pyspark/ml/regression.py b/python/pyspark/ml/regression.py index a0f7f54..e12abeb 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) --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
