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

from dataclasses import dataclass

from immuneML.data_model.dataset.Dataset import Dataset
from immuneML.environment.LabelConfiguration import LabelConfiguration
from immuneML.hyperparameter_optimization.config.SplitConfig import SplitConfig
from immuneML.hyperparameter_optimization.config.SplitType import SplitType
from immuneML.workflows.steps.StepParams import StepParams


[docs]@dataclass class DataSplitterParams(StepParams): dataset: Dataset split_strategy: SplitType split_count: int training_percentage: float = -1 paths: list = None split_config: SplitConfig = None label_config: LabelConfiguration = None