Skip to content
Snippets Groups Projects
Commit 4a6a6578 authored by function2's avatar function2
Browse files

minor change

parent 2d5229ef
Branches
Tags
No related merge requests found
...@@ -5,8 +5,13 @@ import zipfile ...@@ -5,8 +5,13 @@ import zipfile
from convlab2 import DATA_ROOT from convlab2 import DATA_ROOT
def get_subdir(subtask):
subdir = 'multiwoz_zh' if subtask == 'multiwoz' else 'crosswoz_en'
return subdir
def prepare_data(subtask, split, data_root=DATA_ROOT): def prepare_data(subtask, split, data_root=DATA_ROOT):
data_dir = os.path.join(data_root, 'multiwoz_zh' if subtask == 'multiwoz' else 'crosswoz_en') data_dir = os.path.join(data_root, get_subdir(subtask))
zip_filename = os.path.join(data_dir, f'{split}.json.zip') zip_filename = os.path.join(data_dir, f'{split}.json.zip')
test_data = json.load(zipfile.ZipFile(zip_filename).open(f'{split}.json')) test_data = json.load(zipfile.ZipFile(zip_filename).open(f'{split}.json'))
data = {} data = {}
...@@ -131,8 +136,3 @@ def eval_states(gt, pred, subtask): ...@@ -131,8 +136,3 @@ def eval_states(gt, pred, subtask):
'f1': f1, 'f1': f1,
} }
} }
def get_subdir(subtask):
subdir = 'multiwoz_zh' if subtask == 'multiwoz' else 'crosswoz_en'
return subdir
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment