Views
No views yet
1import gymnasium as gym
2import pickle5 as pickle
3from huggingface_sb3 import load_from_hub
4from hf_course_code import evaluate_agent # Code from the course https://huggingface.co/learn/deep-rl-course/unit2/hands-on#the-evaluation-method-
5
6model_pickle = load_from_hub(repo_id="jostyposty/drl-course-unit-02-taxi-v3", filename="q-learning.pkl")
7
8with open(model_pickle, "rb") as f:
9 model = pickle.load(f)
10
11env = gym.make(model["env_id"])
12
13mean_reward, std_reward = evaluate_agent(
14 env,
15 model["max_steps"],
16 model["n_eval_episodes"],
17 model["qtable"],
18 model["eval_seed"],
19)
20result = mean_reward - std_reward
21print(f"Result={result:.2f}, Mean_reward={mean_reward:.2f} +/- {std_reward:.2f}")