diaact_nl_pairs = params['diaact_nl_pairs'] nlg_model = nlg() nlg_model.load_nlg_model(nlg_model_path) nlg_model.load_predefine_act_nl_pairs(diaact_nl_pairs) # load nlg templates agent.set_nlg_model(nlg_model) user_sim.set_nlg_model(nlg_model) ################################################################################ # load trained NLU model ################################################################################ nlu_model_path = params['nlu_model_path'] nlu_model = nlu() nlu_model.load_nlu_model(nlu_model_path) agent.set_nlu_model(nlu_model) user_sim.set_nlu_model(nlu_model) ################################################################################ # Dialog Manager ################################################################################ dialog_manager = DialogManager(agent, user_sim, act_set, slot_set, kb) ################################################################################ # Run num_episodes Conversation Simulations ################################################################################ status = {'successes': 0, 'count': 0, 'cumulative_reward': 0} simulation_epoch_size = params['simulation_epoch_size'] batch_size = params['batch_size'] # default = 16 warm_start = params['warm_start']
diaact_nl_pairs = params['diaact_nl_pairs'] nlg_model = nlg() nlg_model.load_nlg_model(nlg_model_path) nlg_model.load_predefine_act_nl_pairs(diaact_nl_pairs) agent.set_nlg_model(nlg_model) user_sim.set_nlg_model(nlg_model) ################################################################################ # load trained NLU model ################################################################################ nlu_model_path = params['nlu_model_path'] nlu_model = nlu() nlu_model.load_nlu_model(nlu_model_path) agent.set_nlu_model(nlu_model) user_sim.set_nlu_model(nlu_model) ################################################################################ # Dialog Manager ################################################################################ dialog_manager = DialogManager(agent, user_sim, act_set, slot_set, movie_kb) ################################################################################ # Run num_episodes Conversation Simulations ################################################################################ status = {'successes': 0, 'count': 0, 'cumulative_reward': 0} simulation_epoch_size = params['simulation_epoch_size'] batch_size = params['batch_size'] # default = 16 warm_start = params['warm_start']