rl-plotter开发用于绘制强化学习效果图的工作,非常简单好用。
在一般强化学习的论文中效果图都是使用spiningup中相对应的绘制模块完成的,但这相对较麻烦,rl-plotter有效减轻了绘图的工作量。
Github地址:https://github.com/gxywy/rl-plotter
例:
用法(pycharm为例):
1、(Terminal)安装
pip install rl_plotter
2、(.py文件)一般逻辑代码
from rl_plotter.logger import Logger
logger = Logger(exp_name=“your_exp_name”, env_name, seed, locals())
····
logger.update(score=evaluation_score_list, total_steps=current_training_steps)
3、(Terminal)可视化
普通:
rl_plotter --save --show
平均曲线及上下的标准误差阴影区间:
rl_plotter --show --avg_group --shaded_std
平均曲线及与Mean估计误差相对应的阴影区域:
rl_plotter --show --avg_group --shaded_err
Both:
rl_plotter --show --avg_group --shaded_std --shaded_err
看看效果:
如果想要修改图例名称,直接修改生成的csv文件名就好了
rl_plotter --help查看更多命令