在做强化学习时训练是要花费大量的时间的,而训练后的数据是要做到及时保存。这是我做实验时用到的对文件的保存、修改以及打印输出的一段代码。
import pprint, pickle #加载 保存python数据的库
import matplotlib.pyplot as plt #加载图形输出库
pkl_file = open(‘reward.pkl’, ‘rb’) #文件句柄,读打开权限
sum_ep_r = pickle.load(pkl_file) #加载数据,当然这个数据是在之前训练时保存的
pprint.pprint(sum_ep_r) #打印输出结果
pkl_file.close() #关闭文件
如果有些数据有明显的错误,可以做小小的调整。
pkl_file = open(‘reward.pkl’, ‘wb’) #文件句柄,写打开权限
sum_ep_r[2] = -37 #修改内容
pickle.dump(sum_ep_r, pkl_file) #写入到文件中
pprint.pprint(sum_ep_r)
pkl_file.close() #关闭
pkl_file = open(‘reward.pkl’, ‘rb’) #再读出来
sum_ep_r = pickle.load(pkl_file)
pprint.pprint(sum_ep_r)
pkl_file.close()
用图形简单看一下。
plt.figure(1)
plt.plot(range(len(sum_ep_r)),sum_ep_r)
plt.show()