Views
No views yet
1import pickle
2import gymnasium as gym
3from huggingface_hub import hf_hub_download
4
5model_path = hf_hub_download(repo_id="fschnell/q_learning_FrozenLake-v1", filename="q-learning.pkl")
6with open(model_path, "rb") as f:
7 model = pickle.load(f)
8
9env = gym.make(
10 model["env_id"],
11 is_slippery=model.get("is_slippery", True),
12 render_mode="rgb_array",
13)
14q_table = model["qtable"]