(item_names)
| 22 | |
| 23 | |
| 24 | def split_train_test_set(item_names): |
| 25 | item_names = deepcopy(item_names) |
| 26 | test_item_names = [x for x in item_names if any([ts in x for ts in hparams['test_prefixes']])] |
| 27 | train_item_names = [x for x in item_names if x not in set(test_item_names)] |
| 28 | logging.info("train {}".format(len(train_item_names))) |
| 29 | logging.info("test {}".format(len(test_item_names))) |
| 30 | return train_item_names, test_item_names |
| 31 | |
| 32 | |
| 33 | class SingingBinarizer(BaseBinarizer): |