This is an automated email from the ASF dual-hosted git repository.
weichenxu123 pushed a commit to branch fix_pipeline_tuning
in repository https://gitbox.apache.org/repos/asf/spark.git
commit 2af493a016093d8499bf02bc37bc3e842ed74c62
Author: Weichen Xu <weichen...@databricks.com>
AuthorDate: Mon Apr 20 17:47:35 2020 +0800
init
---
python/pyspark/ml/pipeline.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/python/pyspark/ml/pipeline.py b/python/pyspark/ml/pipeline.py
index 0004b64..3a71632 100644
--- a/python/pyspark/ml/pipeline.py
+++ b/python/pyspark/ml/pipeline.py
@@ -210,10 +210,12 @@ class Pipeline(Estimator, MLReadable, MLWritable):
for pair in javaParamMap.toList():
param = pair.param()
if self.hasParam(str(param.name())):
- if param.name() == "classifier":
- paramMap[self.getParam(param.name())] =
JavaParams._from_java(pair.value())
+ java_obj = pair.value()
+ if
sc._jvm.Class.forName("org.apache.spark.ml.PipelineStage").isInstance(java_obj):
+ py_obj = JavaParams._from_java(java_obj)
else:
- paramMap[self.getParam(param.name())] = _java2py(sc,
pair.value())
+ py_obj = _java2py(sc, java_obj)
+ paramMap[self.getParam(param.name())] = py_obj
return paramMap
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@spark.apache.org
For additional commands, e-mail: commits-h...@spark.apache.org
