Xiangrui Meng created SPARK-31777:
-------------------------------------
Summary: CrossValidator supports user-supplied folds
Key: SPARK-31777
URL: https://issues.apache.org/jira/browse/SPARK-31777
Project: Spark
Issue Type: New Feature
Components: ML
Affects Versions: 3.1.0
Reporter: Xiangrui Meng
As a user, I can specify how CrossValidator should create folds by specifying a
foldCol, which should be integer type with range [0, numFolds). If foldCol is
specified, Spark won't do random k-fold split. This is useful if there are
custom logics to create folds, e.g., random split by users instead of random
splits of events.
This is similar to SPARK-16206, which is for the RDD-based APIs.
--
This message was sent by Atlassian Jira
(v8.3.4#803005)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
