Skip to content
Snippets Groups Projects
Commit 6e3746cb authored by function2's avatar function2
Browse files

Revert "update eval"

This reverts commit 62a7d11b.
parent c761fc7b
No related branches found
No related tags found
No related merge requests found
import json
import os import os
import json
import zipfile import zipfile
from copy import deepcopy
from convlab2 import DATA_ROOT from convlab2 import DATA_ROOT
...@@ -22,16 +23,16 @@ def prepare_data(subtask, split, data_root=DATA_ROOT): ...@@ -22,16 +23,16 @@ def prepare_data(subtask, split, data_root=DATA_ROOT):
for i in range(0, len(turns), 2): for i in range(0, len(turns), 2):
sys_utt = turns[i - 1]['text'] if i else None sys_utt = turns[i - 1]['text'] if i else None
user_utt = turns[i]['text'] user_utt = turns[i]['text']
dialog_state = {} state = {}
for domain_name, domain in turns[i + 1]['metadata'].items(): for domain_name, domain in turns[i + 1]['metadata'].items():
if domain_name in ['警察机关', '医院', '公共汽车']: if domain_name in ['警察机关', '医院', '公共汽车']:
continue continue
state = {} domain_state = {}
for slots in domain.values(): for slots in domain.values():
for slot_name, value in slots.items(): for slot_name, value in slots.items():
state[slot_name] = value domain_state[slot_name] = value
dialog_state[domain_name] = state state[domain_name] = domain_state
dialog_data.append((sys_utt, user_utt, dialog_state)) dialog_data.append((sys_utt, user_utt, state))
data[dialog_id] = dialog_data data[dialog_id] = dialog_data
else: else:
for dialog_id, dialog in test_data.items(): for dialog_id, dialog in test_data.items():
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment