Github user WeichenXu123 commented on a diff in the pull request:
https://github.com/apache/spark/pull/19122#discussion_r138518235
--- Diff: python/pyspark/ml/tuning.py ---
@@ -208,23 +210,23 @@ class CrossValidator(Estimator, ValidatorParams,
MLReadable, MLWritable):
@keyword_only
def __init__(self, estimator=None, estimatorParamMaps=None,
evaluator=None, numFolds=3,
- seed=None):
+ seed=None, parallelism=1):
"""
__init__(self, estimator=None, estimatorParamMaps=None,
evaluator=None, numFolds=3,\
- seed=None)
+ seed=None, parallelism=1)
"""
super(CrossValidator, self).__init__()
- self._setDefault(numFolds=3)
+ self._setDefault(numFolds=3, parallelism=1)
--- End diff --
I add check when creating thread pool.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
