torchcell.datasets.scerevisiae.kuzmin2020.SmfKuzmin2020Dataset

class SmfKuzmin2020Dataset(root: str = 'data/torchcell/smf_kuzmin2020', io_workers: int = 0, transform: Callable[[...], Any] | None = None, pre_transform: Callable[[...], Any] | None = None, **kwargs: Any)[source]

Bases: ExperimentDataset

Single-mutant fitness experiments from Kuzmin et al. 2020.

organism: ClassVar[Organism] = 'scerevisiae'

each loader declares its own, the build’s organism switch reads it off the class (dataset_organism), and a loader that forgot it is refused by name instead of inheriting an organism it may not have.

Type:

The host organism every record describes. Annotated here WITHOUT a value

property experiment_class: type[Experiment]

Return the experiment schema class for this dataset.

property reference_class: type[ExperimentReference]

Return the experiment-reference schema class for this dataset.

property raw_file_names: str

Return the raw supplementary Excel file name.

property processed_file_names: str

Return the processed output name (the LMDB directory).

download() → None[source]

Download and extract the supplementary zip, then remove the archive.

process() → None[source]

Read, preprocess, and serialize experiments into the LMDB store.

preprocess_raw(df: DataFrame) → DataFrame[source]

Filter to single mutants and clean allele names for processing.

static create_experiment(dataset_name: str, row: Series) → tuple[FitnessExperiment, FitnessExperimentReference, Publication][source]

Build the experiment, reference, and publication objects from one row.