强化学习RL训练效果图之rl-plotter

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查看更多命令

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值