tabensemb.data.AbstractSplitter._next_cv#
method
- AbstractSplitter._next_cv(df: DataFrame, cont_feature_names: List[str], cat_feature_names: List[str], label_name: List[str], cv: int) Tuple[ndarray, ndarray, ndarray][source]#
Get the next fold of indices of training, validation, and testing sets.
- Parameters:
- df:
The input tabular dataset.
- cont_feature_names:
Names of continuous features.
- cat_feature_names:
Names of categorical features.
- label_name:
The name of the label.
- cv:
The total number of cross-validation runs.
- Returns:
- np.ndarray
Indices of the training, validation, and testing dataset.