aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/ml/tuning.py
diff options
context:
space:
mode:
authorYanbo Liang <ybliang8@gmail.com>2016-05-03 16:46:13 +0200
committerNick Pentreath <nickp@za.ibm.com>2016-05-03 16:46:13 +0200
commitd26f7cb0121767da678bbbbf3a0e31c63d5e3159 (patch)
tree76c2dafc74ec093ac832cf5045fbcf10f9e977e7 /python/pyspark/ml/tuning.py
parent46965cd014fd4ba68bdec15156ec9bcc27d9b217 (diff)
downloadspark-d26f7cb0121767da678bbbbf3a0e31c63d5e3159.tar.gz
spark-d26f7cb0121767da678bbbbf3a0e31c63d5e3159.tar.bz2
spark-d26f7cb0121767da678bbbbf3a0e31c63d5e3159.zip
[SPARK-14971][ML][PYSPARK] PySpark ML Params setter code clean up
## What changes were proposed in this pull request? PySpark ML Params setter code clean up. For examples, ```setInputCol``` can be simplified from ``` self._set(inputCol=value) return self ``` to: ``` return self._set(inputCol=value) ``` This is a pretty big sweeps, and we cleaned wherever possible. ## How was this patch tested? Exist unit tests. Author: Yanbo Liang <ybliang8@gmail.com> Closes #12749 from yanboliang/spark-14971.
Diffstat (limited to 'python/pyspark/ml/tuning.py')
-rw-r--r--python/pyspark/ml/tuning.py6
1 files changed, 2 insertions, 4 deletions
diff --git a/python/pyspark/ml/tuning.py b/python/pyspark/ml/tuning.py
index eb1f029ebb..b21cf92559 100644
--- a/python/pyspark/ml/tuning.py
+++ b/python/pyspark/ml/tuning.py
@@ -198,8 +198,7 @@ class CrossValidator(Estimator, ValidatorParams):
"""
Sets the value of :py:attr:`numFolds`.
"""
- self._set(numFolds=value)
- return self
+ return self._set(numFolds=value)
@since("1.4.0")
def getNumFolds(self):
@@ -350,8 +349,7 @@ class TrainValidationSplit(Estimator, ValidatorParams):
"""
Sets the value of :py:attr:`trainRatio`.
"""
- self._set(trainRatio=value)
- return self
+ return self._set(trainRatio=value)
@since("2.0.0")
def getTrainRatio(self):