torchdms.data.SplitDataset

class torchdms.data.SplitDataset(*, test_data, val_data, train_data_list, description_string)[source]

Bases: object

BinaryMapDatasets for each of test, validation, and train.

Train is partitioned into a list of BinaryMapDatasets according to the number of mutations.

Methods

of_split_df

summarize

Attributes

labeled_splits

Returns an iterator on (label, split) pairs.
property labeled_splits

Returns an iterator on (label, split) pairs.