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. :return: a picklable python function that can be called on a batch by Pytorch loaders

__init__(dataset)

Initialize a Collater object. :return: a picklable python function that can be called on a batch by Pytorch loaders

Methods

__init__(dataset)

Initialize a Collater object.