本项目用于学习RL基础算法,尽量做到: 注释详细,结构清晰。
代码结构主要分为以下几个脚本:
model.py
强化学习算法的基本模型,比如神经网络,actor,critic等memory.py
保存Replay Buffer,用于off-policyplot.py
利用matplotlib或seaborn绘制rewards图,包括滑动平均的reward,结果保存在result文件夹中env.py
用于构建强化学习环境,也可以重新自定义环境,比如给action加noiseagent.py
RL核心算法,比如dqn等,主要包含update和choose_action两个方法,main.py
运行主函数
其中model.py
,memory.py
,plot.py
由于不同算法都会用到,所以放入common
文件夹中。
python 3.7、pytorch 1.6.0-1.7.1、gym 0.17.0-0.18.0
运行带有train
的py文件或ipynb文件进行训练,如果前面带有task
如task0_train.py
,表示对task0任务训练
类似的带有eval
即为测试。