7.3.5 模型训练和测试
本阶段将实型一个整体的训练和测试流程,包含了训练和测试的功能。
(1)定义函数show_state(),其主要功能是在测试过程中显示超级马里奥游戏的环境状态。具体实现代码如下所示。
def show_state(env, ep=0, info=""):
"""在测试时显示超级马里奥游戏环境的状态"""
plt.figure(3)
plt.clf()
plt.imshow(env.render(mode='rgb_array'))
plt.title("Episode: %d %s" % (ep, info))
plt.axis('off')
display.clear_output(wait=True)
display.display(plt.gcf())
函数show_state(env, ep=0, info="")接受如下所示的3个参数: