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

106 lines
2.5 KiB
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-25 11:25:49 +02:00
"book":{
2022-05-25 11:23:26 +02:00
"booked": [],
"people": "",
"day": "",
"seat": ""
},
"semi": {
"name": "",
"e-mail": "",
"price": "",
"stars": "",
"movie": "",
"type": ""
}
2022-05-25 11:25:49 +02:00
},
2022-05-23 21:07:01 +02:00
"police": {
"book": {
"booked": []
},
"semi": {}
},
"hotel": {
"book": {
"booked": [],
"people": "",
"day": "",
"stay": ""
},
"semi": {
"name": "",
"area": "",
"parking": "",
"pricerange": "",
"stars": "",
"internet": "",
"type": ""
}
},
"attraction": {
"book": {
"booked": []
},
"semi": {
"type": "",
"name": "",
"area": ""
}
},
"restaurant": {
"book": {
"booked": [],
"people": "",
"day": "",
"time": ""
},
"semi": {
"food": "",
"pricerange": "",
"name": "",
"area": "",
}
},
"hospital": {
"book": {
"booked": []
},
"semi": {
"department": ""
}
},
"taxi": {
"book": {
"booked": []
},
"semi": {
"leaveAt": "",
"destination": "",
"departure": "",
"arriveBy": ""
}
},
"train": {
"book": {
"booked": [],
"people": ""
},
"semi": {
"leaveAt": "",
"destination": "",
"day": "",
"arriveBy": "",
"departure": ""
}
}
}
return state