From d3e2e202994e063856c192e9fdd0541777b88e0e Mon Sep 17 00:00:00 2001 From: Tommy YU Date: Thu, 11 Feb 2016 18:38:49 -0800 Subject: [SPARK-13153][PYSPARK] ML persistence failed when handle no default value parameter Fix this defect by check default value exist or not. yanboliang Please help to review. Author: Tommy YU Closes #11043 from Wenpei/spark-13153-handle-param-withnodefaultvalue. --- python/pyspark/ml/wrapper.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/pyspark/ml/wrapper.py b/python/pyspark/ml/wrapper.py index d4d48eb215..f8feaa1dfa 100644 --- a/python/pyspark/ml/wrapper.py +++ b/python/pyspark/ml/wrapper.py @@ -79,8 +79,9 @@ class JavaWrapper(Params): for param in self.params: if self._java_obj.hasParam(param.name): java_param = self._java_obj.getParam(param.name) - value = _java2py(sc, self._java_obj.getOrDefault(java_param)) - self._paramMap[param] = value + if self._java_obj.isDefined(java_param): + value = _java2py(sc, self._java_obj.getOrDefault(java_param)) + self._paramMap[param] = value @staticmethod def _empty_java_param_map(): -- cgit v1.2.3