You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@spark.apache.org by me...@apache.org on 2015/10/06 23:58:51 UTC
spark git commit: [SPARK-10957] [ML] setParams changes
quantileProbabilities unexpectly in PySpark's AFTSurvivalRegression
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 <me...@databricks.com>
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 <me...@databricks.com>
Authored: Tue Oct 6 14:58:42 2015 -0700
Committer: Xiangrui Meng <me...@databricks.com>
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: commits-unsubscribe@spark.apache.org
For additional commands, e-mail: commits-help@spark.apache.org