Readme | 中文文档
Dies ist ein einfaches Werkzeug, mit dem Lernkurven für das Verstärkungslernen (RL) problemlos zeichnen können (RL).
von pip
pip install rl_plotter
von Quelle
python setup.py install
1. Fügen Sie Basic Logger hinzu
Fügen Sie unseren Logger in Ihren Bewertungscode hinzu (empfehlen)
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 )Oder verwenden Sie einfach Openai-Spinningup, um sich zu melden (Support)
oder Sie können Openai-Baseline Bench.Monitor verwenden (nicht empfehlen)
env = logger . monitor_env ( env )2. Verfolgen Sie andere Variablen (optional)
Wenn Sie andere Variablen verfolgen möchten, können Sie unseren Custom_logger verwenden:
custom_logger = logger . new_custom_logger ( filename , fieldnames = [ "variable 1" , "variable 2" , ..., "variable n" ])
custom_logger . update ( fieldvalues = variable_value_list , total_steps = current_training_steps )3. Zeichnen Sie die Ergebnisse
Nach dem Training oder wenn Sie Ihren Agenten trainieren, können Sie die Lernkurven auf diese Weise zeichnen:
Wechseln Sie in das übergeordnete Verzeichnis des Protokollverzeichnisses oder zum Multi -Protokoll (Standard: ./)
Befehl ausführen, um zu zeichnen:
rl_plotter --save --show
Sie können auch Seeborn Kernel verwenden, um den gleichen Handlungsstil wie Openai-SpinningUp zu erhalten:
rl_plotter_spinup --save --show
1. häufig verwendete Befehle
rl_plotter --save --show --filter HalfCheetah
rl_plotter --save --show --filter Ant --avg_group --shaded_std
rl_plotter --save --show --filter Swimmer --avg_group --shaded_std --shaded_err
rl_plotter --save --show --filter Walker2d --filename progress.txt --xkey TotalEnvInteracts --ykey AverageEpRet
2. Praktische Beispiele
rl_plotter --show --save --avg_group --shaded_err --shaded_std

rl_plotter --show --save --avg_group --shaded_err --shaded_std --filename q --filters Walker HalfCheetah --ykey bias real_q --yduel --style default --smooth 0

3. spezifischere Verwendung
Sie können alle Parameter finden, die den Stil Ihrer Kurven help benutzerdefiniert werden können
rl_plotter --help
optional arguments:
-h, --help show this help message and exit
--fig_length matplotlib figure length (default: 8)
--fig_width matplotlib figure width (default: 6)
--style matplotlib figure style (default: seaborn)
--title matplotlib figure title (default: None)
--xlabel matplotlib figure xlabel
--xkey x-axis key in csv file (default: l)
--ykey y-axis key in csv file (support multi) (default: r)
--yduel duel y axis (use if has two ykeys)
--ylabel matplotlib figure ylabel
--smooth smooth radius of y axis (default: 10)
--resample if not zero, size of the uniform grid in x direction
to resample onto. Resampling is performed via
symmetric EMA smoothing (see the docstring for
symmetric_ema). Default is zero (no resampling). Note
that if average_group is True, resampling is
necessary; in that case, default value is 512.
(default: 512)
--smooth_step when resampling (i.e. when resample > 0 or
average_group is True), use this EMA decay parameter
(in units of the new grid step). See docstrings for
decay_steps in symmetric_ema or one_sided_ema functions.
(default: 1.0)
--avg_group average the curves in the same group and plot the mean
--shaded_std shaded region corresponding to standard deviation of the group
--shaded_err shaded region corresponding to error in mean estimate of the group
--legend_loc location of legend
--legend_outside place the legend outside of the figure
--borderpad borderpad of legend (default: 0.5)
--labelspacing labelspacing of legend (default: 0.5)
--no_legend_group_num don't show num of group in legend
--time enable this will activate parameters about time
--time_unit parameters about time, x axis time unit (default: h)
--time_interval parameters about time, x axis time interval (default: 1)
--xformat x-axis format
--xlim x-axis limitation (default: None)
--log_dir log dir (default: ./)
--filters filter of dirname
--filename csv filename
--show show figure
--save save figure
--dpi figure dpi (default: 400)
Wenn Sie dieses Repository für Ihre Forschung oder Veröffentlichung verwenden, geben Sie bitte an:
@misc{rl-plotter,
author = {Xiaoyu Gong},
title = {RL-plotter: A plotter for reinforcement learning},
year = {2020},
publisher = {GitHub},
journal = {GitHub repository},
howpublished = {url{https://github.com/gxywy/rl-plotter}},
}
Der Kern dieses Tools ist von Baselines/Plot_util.py und SpinningUp/Plot.py inspiriert.