src.data.base module

Shared abstract class for dataset builders.

class src.data.base.AbstractDataBuilder(num_workers: int, seed: int | None = None)

Bases: ABC

split_data(y, val_size=0.0, test_size=0.2, random_state=None)

Splits (X, y) into train/val/test. - train: (1 - val_size - test_size) - val: val_size - test: test_size