sinergym.utils.callbacks

Custom Callbacks for stable baselines 3 algorithms.

Functions

evaluate_policy(model, env[, ...])

Runs policy for n_eval_episodes episodes and returns average reward.

Classes

LoggerCallback(*args, **kwargs)

Custom callback for plotting additional values in tensorboard.

LoggerEvalCallback(*args, **kwargs)

Callback for evaluating an agent.