aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/ml/param/_shared_params_code_gen.py
diff options
context:
space:
mode:
authorHolden Karau <holden@pigscanfly.ca>2015-10-07 17:50:35 -0700
committerJoseph K. Bradley <joseph@databricks.com>2015-10-07 17:50:35 -0700
commit3aff0866a8601b4daf760d6bf175f68d5a0c8912 (patch)
tree6f9085d991fb04be7d00fb8159692dfe75fbd30f /python/pyspark/ml/param/_shared_params_code_gen.py
parent1bc435ae3afb7a007b8a8ff00dcad4738a9ff055 (diff)
downloadspark-3aff0866a8601b4daf760d6bf175f68d5a0c8912.tar.gz
spark-3aff0866a8601b4daf760d6bf175f68d5a0c8912.tar.bz2
spark-3aff0866a8601b4daf760d6bf175f68d5a0c8912.zip
[SPARK-9774] [ML] [PYSPARK] Add python api for ml regression isotonicregression
Add the Python API for isotonicregression. Author: Holden Karau <holden@pigscanfly.ca> Closes #8214 from holdenk/SPARK-9774-add-python-api-for-ml-regression-isotonicregression.
Diffstat (limited to 'python/pyspark/ml/param/_shared_params_code_gen.py')
-rw-r--r--python/pyspark/ml/param/_shared_params_code_gen.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/python/pyspark/ml/param/_shared_params_code_gen.py b/python/pyspark/ml/param/_shared_params_code_gen.py
index 5b39e5dd4e..45a94e9c32 100644
--- a/python/pyspark/ml/param/_shared_params_code_gen.py
+++ b/python/pyspark/ml/param/_shared_params_code_gen.py
@@ -133,7 +133,10 @@ if __name__ == "__main__":
("thresholds", "Thresholds in multi-class classification to adjust the probability of " +
"predicting each class. Array must have length equal to the number of classes, with " +
"values >= 0. The class with largest value p/t is predicted, where p is the original " +
- "probability of that class and t is the class' threshold.", None)]
+ "probability of that class and t is the class' threshold.", None),
+ ("weightCol", "weight column name. If this is not set or empty, we treat " +
+ "all instance weights as 1.0.", None)]
+
code = []
for name, doc, defaultValueStr in shared:
param_code = _gen_param_header(name, doc, defaultValueStr)