def log_value_string(self, value_string): """ :description: prints a string to a file. The string, when formatted, gives the values of different states in the mdp. """ if self.log_dir is None: self.create_log_dir() filename = 'value_image.txt' filepath = os.path.join(self.log_dir, filename) with open(filepath, 'wb') as f: f.write(value_string) learning_utils.make_heat_map(filepath, self.epoch)
def test_make_heat_map(self): filepath = '/Users/wulfe/Dropbox/School/Stanford/winter_2016/cs239/project/hierarchical_rl/logs/rqn_4_step_stacked_2roomx5x5_row_col/value_image.txt' epoch = 1 learning_utils.make_heat_map(filepath, epoch)