aitech-sd-lab/DST_DP_lab_9-10/dialogue_state.py

27 lines
664 B
Python
Raw Normal View History

2022-05-23 21:07:01 +02:00
def default_state():
state = dict(user_action=[],
system_action=[],
belief_state={},
request_state={},
terminated=False,
history=[])
state['belief_state'] = {
2022-05-25 11:23:26 +02:00
"cinema": {
2022-05-28 14:18:38 +02:00
"book": {
2022-05-25 11:23:26 +02:00
"booked": [],
2022-05-28 14:18:38 +02:00
"date": "",
"hour": "",
"seat": "",
"movie": ""
2022-05-25 11:23:26 +02:00
},
"semi": {
"name": "",
"e-mail": "",
"price": "",
2022-05-28 14:18:38 +02:00
"number": "",
2022-05-25 11:23:26 +02:00
"type": ""
}
2022-05-23 21:07:01 +02:00
}
}
return state