17.8.6 评估自动驾驶模型的性能
编写文件test_agent.py,功能是测试一个已经训练好的自动驾驶智能体的性能,以便了解它在给定路线和环境条件下的表现如何。可以通过命令行参数指定要测试的路线、智能体模型文件和其他配置信息。与本节前面中提到的训练脚本相比,这个脚本的目标是测试已经训练好的智能体,而不是训练新的智能体模型。具体实现代码如下所示。
def test_agent(env, weather_list, agent, route_id, nb_episodes):
ep_rewards = []
success_rate = 0
avg_steps = 0
for episode in range(nb_episodes):
weather = weather_list[episode%len(weather_list)]
env.set_weather(weather)
obs = env.reset(route_id)
done = False
episode_reward = 0
nb_steps = 0
while not done:
act = agent.select_action(obs, noise&