moleculib.sequence.dataset¶

Classes¶

`ElutedLigandDataset`
`MHCBindingAffinityDataset`
`GFPFitnessDataset`

Functions¶

onehot_encode_datum(sequencedatum)

Module Contents¶

moleculib.sequence.dataset.onehot_encode_datum(sequencedatum: moleculib.sequence.datum.SeqDatum)¶

class moleculib.sequence.dataset.ElutedLigandDataset(base_path: str, transform: moleculib.protein.transform.ProteinTransform = None, encode_one_hot: bool = True)¶

Bases: torch.utils.data.Dataset

base_path¶

transform = None¶

encode_one_hot = True¶

data_frame = None¶

splits¶

class moleculib.sequence.dataset.MHCBindingAffinityDataset(base_path: str, transform: moleculib.protein.transform.ProteinTransform = None)¶

Bases: torch.utils.data.Dataset

base_path¶

transform = None¶

data_frame = None¶

splits¶

class moleculib.sequence.dataset.GFPFitnessDataset(base_path: str, transform: moleculib.protein.transform.ProteinTransform = None, encode_one_hot: bool = True, encode_lm: bool = False, lm_model: str = 'esm')¶

Bases: torch.utils.data.Dataset

base_path¶

transform = None¶

data_frame = None¶

splits¶

encode_one_hot = True¶