Example #1
0
    def update(self, model1, model2):
        """
        Args:
            model1 - tuple containing (experiment, version, step) for the first
            compared model model2 - tuple containing (experiment, version,
            step) for the second compared model.
        """

        success_map_1 = DataReader.get_episode_success_map(*model1)
        success_map_2 = DataReader.get_episode_success_map(*model2)

        result = success_map_1 + 2 * success_map_2
        self.result_permutation = np.argsort(result)
        result = np.sort(result)
        self.episode_grid_heat_map.color = result.reshape(11, 51)
        self.color_result = result