# -*- 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)