Source code for immuneML.workflows.steps.data_splitter.Util

from immuneML.data_model.dataset.Dataset import Dataset
from immuneML.util.PathBuilder import PathBuilder
from immuneML.workflows.steps.data_splitter.DataSplitterParams import DataSplitterParams


[docs]class Util:
[docs] @staticmethod def prepare_path(input_params: DataSplitterParams, split_index: int) -> str: path = input_params.paths[split_index] / "datasets" PathBuilder.build(path) return path
[docs] @staticmethod def make_dataset(dataset: Dataset, indices, input_params: DataSplitterParams, i: int, dataset_type: str): path = Util.prepare_path(input_params, i) new_dataset = dataset.make_subset(indices, path, dataset_type) return new_dataset