torchcell.datasets.OneHotGeneDataset

class OneHotGeneDataset(root: str, genome: SCerevisiaeGenome, transform: Callable[[...], Any] | None = None, pre_transform: Callable[[...], Any] | None = None)

Bases: BaseEmbeddingDataset

Embedding dataset producing one-hot vectors over the genome gene set.

static parse_genome(genome: SCerevisiaeGenome | None) → ParsedGenome | None

Return a ParsedGenome holding the gene set, or None if genome is None.

initialize_model() → None

Return None; one-hot encoding requires no backing model.

one_hot_encode_gene(gene: str) → Tensor

One-hot encode a gene based on its position in the gene set.

process() → None

Encode every gene as a one-hot Data object and save the collated dataset.