Github user MrBago commented on a diff in the pull request:
https://github.com/apache/spark/pull/20058#discussion_r159004397
--- Diff: python/pyspark/ml/base.py ---
@@ -47,6 +74,24 @@ def _fit(self, dataset):
"""
raise NotImplementedError()
+ @since("2.3.0")
+ def fitMultiple(self, dataset, params):
+ """
+ Fits a model to the input dataset for each param map in params.
+
+ :param dataset: input dataset, which is an instance of
:py:class:`pyspark.sql.DataFrame`.
+ :param params: A list/tuple of param maps.
--- End diff --
I changed the docstring to `Sequence` instead of `list/tuple`, is that ok?
Do you want to explicitly restrict the input to be a list or tuple?
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
