aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/ml/tuning.py
diff options
context:
space:
mode:
authorYanbo Liang <ybliang8@gmail.com>2016-04-20 13:02:37 -0700
committerJoseph K. Bradley <joseph@databricks.com>2016-04-20 13:02:37 -0700
commit296c384aff5bc1c0e8d411669f8f3e082cdf8f55 (patch)
treeb48a6cef7cd90fb63a597beea99a4b6c0453fd97 /python/pyspark/ml/tuning.py
parent8fc267ab3322e46db81e725a5cb1adb5a71b2b4d (diff)
downloadspark-296c384aff5bc1c0e8d411669f8f3e082cdf8f55.tar.gz
spark-296c384aff5bc1c0e8d411669f8f3e082cdf8f55.tar.bz2
spark-296c384aff5bc1c0e8d411669f8f3e082cdf8f55.zip
[MINOR][ML][PYSPARK] Fix omissive params which should use TypeConverter
## What changes were proposed in this pull request? #11663 adds type conversion functionality for parameters in Pyspark. This PR find out the omissive ```Param``` that did not pass corresponding ```TypeConverter``` argument and fix them. After this PR, all params in pyspark/ml/ used ```TypeConverter```. ## How was this patch tested? Existing tests. cc jkbradley sethah Author: Yanbo Liang <ybliang8@gmail.com> Closes #12529 from yanboliang/typeConverter.
Diffstat (limited to 'python/pyspark/ml/tuning.py')
-rw-r--r--python/pyspark/ml/tuning.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/python/pyspark/ml/tuning.py b/python/pyspark/ml/tuning.py
index ef14da488e..b16628bc70 100644
--- a/python/pyspark/ml/tuning.py
+++ b/python/pyspark/ml/tuning.py
@@ -448,7 +448,7 @@ class TrainValidationSplit(Estimator, ValidatorParams, MLReadable, MLWritable):
"""
trainRatio = Param(Params._dummy(), "trainRatio", "Param for ratio between train and\
- validation data. Must be between 0 and 1.")
+ validation data. Must be between 0 and 1.", typeConverter=TypeConverters.toFloat)
@keyword_only
def __init__(self, estimator=None, estimatorParamMaps=None, evaluator=None, trainRatio=0.75,