Permalink
Switch branches/tags
Nothing to show
Find file Copy path
0eb3b5e Feb 2, 2018
0 contributors

Users who have contributed to this file

11 lines (8 sloc) 412 Bytes
include("src/base.jl")
include("src/envs/toytext_mdp.jl")
include("src/policies/value_table_policy.jl")
include("src/solvers/sarsa_policy_solver.jl")
gym_env = ToyTextMDP(gym.make("Taxi-v2"))
solver = SarsaPolicySolver(5000; max_frame_iterations = 100, epsilon = 0.25, epsilon_discount = 0.999, learning_rate = 0.15, discount = 0.99)
policy = solve(solver, gym_env, verbose = true)
# println(policy.value_map)