diff options
Diffstat (limited to 'python/pyspark/ml/param/_shared_params_code_gen.py')
| -rw-r--r-- | python/pyspark/ml/param/_shared_params_code_gen.py | 4 |
1 files changed, 3 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 c32dcc467d..4f4328bcad 100644 --- a/python/pyspark/ml/param/_shared_params_code_gen.py +++ b/python/pyspark/ml/param/_shared_params_code_gen.py @@ -147,7 +147,9 @@ if __name__ == "__main__": ("solver", "the solver algorithm for optimization. If this is not set or empty, " + "default value is 'auto'.", "'auto'", "TypeConverters.toString"), ("varianceCol", "column name for the biased sample variance of prediction.", - None, "TypeConverters.toString")] + None, "TypeConverters.toString"), + ("aggregationDepth", "suggested depth for treeAggregate (>= 2).", "2", + "TypeConverters.toInt")] code = [] for name, doc, defaultValueStr, typeConverter in shared: |