rnaglib.dataset_transforms.NameSplitter

class rnaglib.dataset_transforms.NameSplitter(train_names, val_names, test_names, **kwargs)

Splits a dataset based on hard-coded lists of RNA names to be included in train, val and test sets

Parameters:

• train_names (list[str]) – list of RNAs to include in the train set

• val_names (list[str]) – list of RNAs to include in the val set

• test_names (list[str]) – list of RNAs to include in the test set

__init__(train_names, val_names, test_names, **kwargs)

Methods

__init__(train_names, val_names, test_names, ...)
forward(dataset)	Apply the splitting to a dataset