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)
