rnaglib.dataset_transforms.Splitter

class rnaglib.dataset_transforms.Splitter(split_train=0.7, split_valid=0.15, split_test=0.15, debug=False)

Objects enabling the splitting of an RNADataset into train, validation and test sets

Parameters:

• split_train (float) – proportion of the dataset to include in the train set (default 0.7)

• split_valid (float) – proportion of the dataset to include in the validation set (default 0.15)

• split_test (float) – proportion of the dataset to include in the test set (default 0.15)

• debug (bool) – whether to run the splitting in debug mode (default False)

__init__(split_train=0.7, split_valid=0.15, split_test=0.15, debug=False)

Methods

__init__([split_train, split_valid, ...])
forward(dataset)