torchcell.datamodules.DatasetSplit

class DatasetSplit(*, phenotype_label_index: dict[str, IndexSplit] | None = None, perturbation_count_index: dict[int, IndexSplit] | None = None, dataset_name_index: dict[str, IndexSplit] | None = None)

Bases: BaseModel

Per-split index groupings keyed by phenotype, perturbation count, or dataset.

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].