Views
No views yet
1import gymnasium as gym
2from huggingface_sb3 import load_from_hub
3import numpy as np
4import pickle
5
6# Load the Q table
7env_name = "FrozenLake-v1"
8model_name = "q-FrozenLake-v1-4x4-noSlippery"
9model_path = load_from_hub(repo_id="ch-bz/" + model_name, filename="q-learning.pkl")
10Qtable = pickle.load(open(model_path, "rb"))["qtable"]
11
12# Run the demonstration of the result
13env = gym.make("FrozenLake-v1", map_name="4x4", is_slippery=False, render_mode="human")
14state, info = env.reset()
15
16while True:
17 action = np.argmax(Qtable[state][:])
18 state, reward, terminated, truncated, info = env.step(action)
19 env.render()
20
21 if terminated or truncated:
22 state, info = env.reset()