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