From ae558fd5f273216c1a5211e82705fe77790702fd Mon Sep 17 00:00:00 2001 From: zqwerty <zhuq96@hotmail.com> Date: Tue, 2 Aug 2022 09:24:58 +0800 Subject: [PATCH] merge seen&unseen testset of wow --- convlab/base_models/t5/key2gen/create_data.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/convlab/base_models/t5/key2gen/create_data.py b/convlab/base_models/t5/key2gen/create_data.py index cb4e12c8..138808f2 100644 --- a/convlab/base_models/t5/key2gen/create_data.py +++ b/convlab/base_models/t5/key2gen/create_data.py @@ -83,6 +83,9 @@ def create_personachat_data(dataset, data_dir, args): def create_wow_data(dataset, data_dir, args): data_by_split = dataset os.makedirs(data_dir, exist_ok=True) + data_by_split['test'] = data_by_split['test_seen'] + data_by_split['test_unseen'] + data_by_split.pop('test_seen') + data_by_split.pop('test_unseen') data_splits = data_by_split.keys() for data_split in data_splits: -- GitLab