seqgra.learner.torch.torchdataset module¶

MIT - CSAIL - Gifford Lab - seqgra

PyTorch DataSet class

@author: Konstantin Krismer

class IterableMultiClassDataSet(file_name: str, learner: seqgra.learner.learner.Learner, shuffle: bool = False, contains_y: bool = True, cache_size: int = 10000)[source]¶: Bases: torch.utils.data.dataset.Dataset[torch.utils.data.dataset.T_co]

class IterableMultiLabelDataSet(file_name: str, learner: seqgra.learner.learner.Learner, shuffle: bool = False, contains_y: bool = True, cache_size: int = 10000)[source]¶: Bases: torch.utils.data.dataset.Dataset[torch.utils.data.dataset.T_co]

class MultiClassDataSet(x, y=None)[source]¶: Bases: Generic[torch.utils.data.dataset.T_co]

class MultiLabelDataSet(x, y=None)[source]¶: Bases: Generic[torch.utils.data.dataset.T_co]