cuml.model_selection#

This code is developed and maintained by scikit-learn and imported by cuML to maintain the familiar sklearn namespace structure. cuML includes tests to ensure full compatibility of these wrappers with CUDA-based data and cuML estimators, but all of the underlying code is due to the scikit-learn developers.

train_test_split

Split arrays or matrices into random train and test subsets.

KFold

K-Folds cross-validator.