def test_value(self, input_param, result): partitions = partition_dataset(**input_param) train = select_cross_validation_folds(partitions=partitions, folds=[0] + list(range(2, len(partitions)))) self.assertListEqual(train, result[0]) val = select_cross_validation_folds(partitions=partitions, folds=1) self.assertListEqual(val, result[1])
def _split_datalist(self, datalist: List[Dict]) -> List[Dict]: data = partition_dataset(data=datalist, num_partitions=nfolds, shuffle=True, seed=seed) return select_cross_validation_folds(partitions=data, folds=folds)