Exemplo n.º 1
0
    def run_loaded_agent(self):
        input_width = 3
        input_height = 4
        n_actions = 2
        max_iter = 100
        discount = 0.9
        learn_rate = .005
        batch_size = 4
        rng = np.random
        filename = "agent_max_iter-{}-width-{}-height-{}-discount-{}-lr-{}-batch-{}.npz".format(max_iter,
                                                                                                input_width,
                                                                                                input_height,
                                                                                                discount,
                                                                                                learn_rate,
                                                                                                batch_size)

        agent_obj = DeepQLearner(input_width, input_height, n_actions, discount, learn_rate, batch_size, rng)

        try:
            agent_obj.load(filename)
        except:
            print "Failed to Load file. Aborting."
            return

        self.test_agent(agent_obj, input_height, input_width)