功能超全的强化学习画图脚本
相比于原始的Spinning up 的plot.py文件
原始画图效果:
线条多一点,就根本分不清谁是谁了。
加了八大功能:
- 可以直接在pycharm或者vscode执行,也可以用命令行传参;
- 按exp_name排序,而不是按时间排序;
- 固定好每个exp_name的颜色;
- 可以调节曲线的线宽,便于观察;
- 保存图片到本地,便于远程ssh画图~
- 自动显示全屏
- 图片自适应
- 针对颜色不敏感的人群,可以在每条legend上注明性能值,和性能序号
seaborn版本0.8.1
最终画图效果:
如果大家还有什么新的功能,我也可以想办法加上去~
代码链接,有详细注释:
https://github.com/kaixindelele/DRLib/blob/main/spinup_utils/plot.py
联系方式:
ps: 欢迎做强化的同学加群一起学习:
欢迎关注知乎帐号:未入门的炼丹学徒
CSDN帐号:https://blog.csdn.net/hehedadaq
极简spinup+HER+PER代码实现:https://github.com/kaixindelele/DRLib