tabensemb.data.AbstractSplitter._sklearn_k_fold

method

AbstractSplitter._sklearn_k_fold(data, cv) → Tuple[ndarray, ndarray, ndarray]

Generate a sklearn.model_selection.KFold instance and return its __next__() result.

Parameters:

data

An Iterable whose index will be split by KFold.

cv

The total number of cross-validation runs.

Returns:

np.ndarray

Indices of the training, validation, and testing datasets of the current fold.

Notes

The returned values are fold indices of the input data argument, but not the fold.