diff options
author | Yanbo Liang <ybliang8@gmail.com> | 2016-04-20 13:02:37 -0700 |
---|---|---|
committer | Joseph K. Bradley <joseph@databricks.com> | 2016-04-20 13:02:37 -0700 |
commit | 296c384aff5bc1c0e8d411669f8f3e082cdf8f55 (patch) | |
tree | b48a6cef7cd90fb63a597beea99a4b6c0453fd97 /python/pyspark | |
parent | 8fc267ab3322e46db81e725a5cb1adb5a71b2b4d (diff) | |
download | spark-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')
-rw-r--r-- | python/pyspark/ml/regression.py | 7 | ||||
-rw-r--r-- | python/pyspark/ml/tuning.py | 2 |
2 files changed, 5 insertions, 4 deletions
diff --git a/python/pyspark/ml/regression.py b/python/pyspark/ml/regression.py index da74ab5070..8e76070e9a 100644 --- a/python/pyspark/ml/regression.py +++ b/python/pyspark/ml/regression.py @@ -561,7 +561,7 @@ class TreeRegressorParams(Params): impurity = Param(Params._dummy(), "impurity", "Criterion used for information gain calculation (case-insensitive). " + "Supported options: " + - ", ".join(supportedImpurities)) + ", ".join(supportedImpurities), typeConverter=TypeConverters.toString) def __init__(self): super(TreeRegressorParams, self).__init__() @@ -1261,11 +1261,12 @@ class GeneralizedLinearRegression(JavaEstimator, HasLabelCol, HasFeaturesCol, Ha family = Param(Params._dummy(), "family", "The name of family which is a description of " + "the error distribution to be used in the model. Supported options: " + - "gaussian(default), binomial, poisson and gamma.") + "gaussian(default), binomial, poisson and gamma.", + typeConverter=TypeConverters.toString) link = Param(Params._dummy(), "link", "The name of link function which provides the " + "relationship between the linear predictor and the mean of the distribution " + "function. Supported options: identity, log, inverse, logit, probit, cloglog " + - "and sqrt.") + "and sqrt.", typeConverter=TypeConverters.toString) @keyword_only def __init__(self, labelCol="label", featuresCol="features", predictionCol="prediction", 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, |