tabensemb.data.datasplitter.RandomSplitter#

class tabensemb.data.datasplitter.RandomSplitter(train_val_test: List | ndarray | None = None, cv: int = -1)[source]#

Bases: AbstractSplitter

Randomly split the dataset.

Methods

__init__(train_val_test: List | ndarray | None = None, cv: int = -1)#

_next_cv(df, cont_feature_names, ...)

Get the next fold of indices of training, validation, and testing sets.

_split(df, cont_feature_names, ...)