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