# -*- coding: utf-8 -*- """ Created on Thu Apr 26 00:02:49 2018 @author: Shiratori """ import graphics as gfx import pong_model as pm from q_learning import QLearning if __name__ == '__main__': # Set up environment environment = pm.PongModel(0.5, 0.5, -0.03, 0.01, 0.4, paddleX=0) window = gfx.GFX(wall_x=1, player_x=0) window.fps = 9e16 # Set up model model = QLearning(environment, window, C=5e3, gamma=0.99, explore=-1, threshold=-1, log=True, log_file='q_test_log_(agent_B).txt', mode='test', q_table_file='q_q_table_(agent_B).csv') # Training model.train()
def main(): env = gym.make('FrozenLake-v0') model = QLearning(env, epsilon=0.2, gamma=1, alpha=0.01) model.train(1000000) model.test(10, render=False)