tumourkit.classification.train_graphs.load_dataset

tumourkit.classification.train_graphs.load_dataset(train_node_dir: str, val_node_dir: str, test_node_dir: str, bsize: int, remove_prior: bool | None = False, remove_morph: bool | None = False, enable_background: bool | None = False) → Tuple[GraphDataLoader, GraphDataLoader, GraphDataLoader]

Creates Torch dataloaders for training. Folder structure: node_dir:

• train

• graphs

• file1.nodes.csv

…

• validation

• graphs

• file1.nodes.csv

…

• test

• graphs

• file1.nodes.csv

…