This commit is contained in:
Tilman Kamp 2019-09-17 10:30:19 +02:00
Родитель eba1385523
Коммит 72b8cc45f4
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -266,7 +266,7 @@ def main(args):
else:
train_size, sample_size = get_set_sizes(len(partition_fragments))
if args.split_field:
portions = engroup(partition_fragments, lambda f: get_meta(f, args.split_field)).values()
portions = list(engroup(partition_fragments, lambda f: get_meta(f, args.split_field)).values())
portions.sort(key=lambda p: len(p))
train_set, dev_set, test_set = [], [], []
for offset, sample_set in [(0, dev_set), (1, test_set)]: