enerzyme.tasks.splitter.Splitter

class enerzyme.tasks.splitter.Splitter(method: Literal['random'] = 'random', seed: int = 114514, preload: bool = True, save: bool = True, **params)

Bases: object

__init__(method: Literal['random'] = 'random', seed: int = 114514, preload: bool = True, save: bool = True, **params)

get_split(data: Dict[str, FieldDataset], preload_path: Dict[str, str] | None = None) → Dict[str, Dict[str, List[int]]]

preload_split(preload_path: Dict[str, str] | None) → bool