Skip to content

Commit d3e2e20

Browse files
Wenpeimengxr
authored andcommitted
[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 <[email protected]> Closes #11043 from Wenpei/spark-13153-handle-param-withnodefaultvalue.
1 parent 5f1c359 commit d3e2e20

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

python/pyspark/ml/wrapper.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,8 +79,9 @@ def _transfer_params_from_java(self):
7979
for param in self.params:
8080
if self._java_obj.hasParam(param.name):
8181
java_param = self._java_obj.getParam(param.name)
82-
value = _java2py(sc, self._java_obj.getOrDefault(java_param))
83-
self._paramMap[param] = value
82+
if self._java_obj.isDefined(java_param):
83+
value = _java2py(sc, self._java_obj.getOrDefault(java_param))
84+
self._paramMap[param] = value
8485

8586
@staticmethod
8687
def _empty_java_param_map():

0 commit comments

Comments
 (0)