From 4053630553e6341bf2cabe260f80a11c9af4cd5b Mon Sep 17 00:00:00 2001 From: Xiangrui Meng Date: Tue, 6 Oct 2015 12:57:40 -0700 Subject: [PATCH] remove special handling for quantileProbabilities --- python/pyspark/ml/regression.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/python/pyspark/ml/regression.py b/python/pyspark/ml/regression.py index a0f7f54e65213..e12abeba019f7 100644 --- a/python/pyspark/ml/regression.py +++ b/python/pyspark/ml/regression.py @@ -701,11 +701,7 @@ def setParams(self, featuresCol="features", labelCol="label", predictionCol="pre 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)