movie_request_slots, movie_inform_slots)
elif agt == 5: # movie request rule agent
agent = RequestBasicsAgent(kb, act_set, slot_set, agent_params,
movie_request_slots)
elif agt == 6: # restaurant request rule agent
agent = RequestBasicsAgent(kb, act_set, slot_set, agent_params,
restaurant_request_slots)
elif agt == 7: # taxi request agent
agent = RequestBasicsAgent(kb, act_set, slot_set, agent_params,
taxi_request_slots)
elif agt == 8: # taxi request-inform rule agent
agent = RequestInformSlotAgent(kb, act_set, slot_set, agent_params,
taxi_request_slots, taxi_inform_slots)
elif agt == 9: # DQN agent for movie domain
agent = AgentDQN(kb, act_set, slot_set, agent_params)
agent.initialize_config(movie_request_slots, movie_inform_slots)
elif agt == 10: # restaurant request-inform rule agent
agent = RequestInformSlotAgent(kb, act_set, slot_set, agent_params,
restaurant_request_slots,
restaurant_inform_slots)
elif agt == 11: # taxi request-inform-cost rule agent
agent = RequestInformSlotAgent(kb, act_set, slot_set, agent_params,
taxi_request_slots, taxi_inform_cost_slots)
elif agt == 12: # DQN agent for restaurant domain
load_actions(dialog_config.restaurant_sys_request_slots,
dialog_config.restaurant_sys_inform_slots)
agent = AgentDQN(kb, act_set, slot_set, agent_params)
agent.initialize_config(restaurant_request_slots, restaurant_inform_slots)
elif agt == 13: # DQN agent for taxi domain
load_actions(dialog_config.taxi_sys_request_slots,
dialog_config.taxi_sys_inform_slots)
agent = EchoAgent(kb, act_set, slot_set, agent_params)
elif agt == 5: # movie request rule agent
agent = RequestBasicsAgent(kb, act_set, slot_set, agent_params,
movie_request_slots)
elif agt == 6: # restaurant request rule agent
agent = RequestBasicsAgent(kb, act_set, slot_set, agent_params,
restaurant_request_slots)
elif agt == 7: # taxi request agent
agent = RequestBasicsAgent(kb, act_set, slot_set, agent_params,
taxi_request_slots)
elif agt == 8: # taxi request-inform rule agent
agent = RequestInformSlotAgent(kb, act_set, slot_set, agent_params,
taxi_request_slots, taxi_inform_slots)
elif agt == 9: # DQN agent for movie domain
agent = AgentDQN(kb, act_set, slot_set, agent_params)
agent.initialize_config(dialog_config.movie_request_slots,
dialog_config.movie_inform_slots)
elif agt == 10: # restaurant request-inform rule agent
agent = RequestInformSlotAgent(kb, act_set, slot_set, agent_params,
restaurant_request_slots,
restaurant_inform_slots)
elif agt == 11: # taxi request-inform-cost rule agent
agent = RequestInformSlotAgent(kb, act_set, slot_set, agent_params,
taxi_request_slots, taxi_inform_cost_slots)
elif agt == 12: # DQN agent for restaurant domain
load_actions(dialog_config.restaurant_sys_request_slots,
dialog_config.restaurant_sys_inform_slots)
agent = AgentDQN(kb, act_set, slot_set, agent_params)
agent.initialize_config(dialog_config.restaurant_request_slots,
dialog_config.restaurant_inform_slots)
elif agt == 13: # DQN agent for taxi domain
load_actions(dialog_config.taxi_sys_request_slots,
