Skip to content
Snippets Groups Projects
lecture_02_frozen_lake.py 622 B
Newer Older
  • Learn to ignore specific revisions
  • tuhe's avatar
    tuhe committed
    # This file may not be shared/redistributed without permission. Please read copyright notice in the git repo. If this file contains other copyright notices disregard this text.
    from irlc.gridworld.gridworld_environments import FrozenLake
    from gymnasium.wrappers import TimeLimit
    from irlc import Agent, interactive, train
    
    if __name__ == "__main__":
        env = FrozenLake(is_slippery=True, living_reward=-1e-4, render_mode="human")
        N = 40
        env, agent = interactive(env, Agent(env))
        env = TimeLimit(env, max_episode_steps=N)
        num_episodes = 100
        train(env, agent, num_episodes=num_episodes)
        env.close()