예제 #1
0
    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)
예제 #2
0
    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)
 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)