rnaglib.dataset_transforms.get_loader

rnaglib.dataset_transforms.get_loader(dataset, batch_size=5, num_workers=0, split=True, split_train=0.7, split_valid=0.85, verbose=False, persistent_workers=True)[source]

Fetch a loader object for a given dataset.

Parameters:

  • dataset (rnaglib.dataset.RNADataset) – Dataset for loading.

  • batch_size (int) – number of items in batch

  • split (bool) – whether to compute splits

  • split_train (float) – proportion of dataset to keep for training

  • split_valid (float) – proportion of dataset to keep for validation

  • verbose (bool) – print updates

Returns:

torch.utils.data.DataLoader