agent_params['warm_start'] = params['warm_start']
agent_params['cmd_input_mode'] = params['cmd_input_mode']
if agt == 0:
agent = AgentCmd(course_kb, act_set, slot_set, agent_params)
elif agt == 1:
agent = InformAgent(course_kb, act_set, slot_set, agent_params)
elif agt == 2:
agent = RequestAllAgent(course_kb, act_set, slot_set, agent_params)
elif agt == 3:
agent = RandomAgent(course_kb, act_set, slot_set, agent_params)
elif agt == 4:
agent = EchoAgent(course_kb, act_set, slot_set, agent_params)
elif agt == 5:
agent = RequestBasicsAgent(course_kb, act_set, slot_set, agent_params)
elif agt == 9:
agent = AgentDQN(course_kb, act_set, slot_set, agent_params)
################################################################################
# Add your agent here
################################################################################
else:
pass
################################################################################
# Parameters for User Simulators
################################################################################
usersim_params = {}
usersim_params['max_turn'] = max_turn
usersim_params['slot_err_probability'] = params['slot_err_prob']
agent_params['cmd_input_mode'] = params['cmd_input_mode']
if agt == 0:
agent = AgentCmd(kb, act_set, slot_set, agent_params)
elif agt == 1:
agent = InformAgent(kb, act_set, slot_set, agent_params)
elif agt == 2:
agent = RequestAllAgent(kb, act_set, slot_set, agent_params)
elif agt == 3:
agent = RandomAgent(kb, act_set, slot_set, agent_params)
elif agt == 4:
#agent = EchoAgent(kb, act_set, slot_set, agent_params)
agent = RequestInformSlotAgent(kb, act_set, slot_set, agent_params,
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,
agent_params['trained_model_path'] = params['trained_model_path']
agent_params['warm_start'] = params['warm_start']
agent_params['cmd_input_mode'] = params['cmd_input_mode']
if agt == 0:
agent = AgentCmd(movie_kb, act_set, slot_set, agent_params)
elif agt == 1:
agent = InformAgent(movie_kb, act_set, slot_set, agent_params)
elif agt == 2:
agent = RequestAllAgent(movie_kb, act_set, slot_set, agent_params)
elif agt == 3:
agent = RandomAgent(movie_kb, act_set, slot_set, agent_params)
elif agt == 4:
agent = EchoAgent(movie_kb, act_set, slot_set, agent_params)
elif agt == 5:
agent = RequestBasicsAgent(movie_kb, act_set, slot_set, agent_params)
elif agt == 9:
agent = AgentDQN(movie_kb, act_set, slot_set, agent_params)
################################################################################
# Add your agent here
################################################################################
else:
pass
################################################################################
# Parameters for User Simulators
################################################################################
usersim_params = {}
usersim_params['max_turn'] = max_turn
usersim_params['slot_err_probability'] = params['slot_err_prob']