cellmil.utils.train.dataset

Functions

complementary_frequencies(df[, label])
split_dataset(df[, train_size, test, ...])

cellmil.utils.train.dataset.split_dataset(df: DataFrame, train_size: float = 0.8, test: bool = False, stratify: bool = True, label: str = 'label', random_state: int = 42)

cellmil.utils.train.dataset.complementary_frequencies(df: DataFrame, label: str = 'label') → dict[int, float]