rnaglib.dataset_transforms.Collater

class rnaglib.dataset_transforms.Collater(dataset)

Wrapper for collate function, so we can use different node similarities. We cannot use functools.partial as it is not picklable so incompatible with Pytorch loading

Initialize a Collater object.

Parameters:

dataset – RNADataset to use for collating batches

__init__(dataset)

Initialize a Collater object.

Parameters:

dataset – RNADataset to use for collating batches

Methods

__init__(dataset)

Initialize a Collater object.