def run(self):
"""
Run experiment
"""
# Coordination experiment
self.expt_name = "expt_00"
self.config['RL_parameters']['experiment'] = self.expt_name
self.config['RL_parameters'][
'city_states_filename'] = "city_states.dill"
self.config['RL_parameters'][
'best_model_filename'] = "expt_00_model.dill"
self.config['RL_parameters'][
'best_episode_filename'] = "expt_00_episode.dill"
self.config['RL_parameters'][
'training_tracker_filename'] = "expt_00_training_tracker.dill"
self.config['RL_parameters']['allow_coordination'] = True
# Hyperparameter values
self.config['RL_parameters']['ind_episodes'] = 60
self.config['RL_parameters']['reb_episodes'] = 160
self.config['RL_parameters']['discount_factor'] = 0.99
self.logger.info("Starting expt_00 coordination")
rl_trainer = RunRLTrainingJob(self.config)
rl_trainer.run()
self.logger.info("Finished expt_00 coordination")
def run_rl_training(config):
rl_trainer = RunRLTrainingJob(config)
try:
data = rl_trainer.run()
except BaseException:
print config
raise ValueError
return data
def run_rl_training(config):
rl_trainer = RunRLTrainingJob(config)
data = rl_trainer.run()
return data