Trajectory Callback Class

class or_suite.experiment.trajectory_callback.TrajectoryCallback(verbose=0)

Custom callback for plotting trajectory information for a StatebleBaselines experiment.

Parameters

verbose – (int) 1 to print trajectory information to console, 0 not to print

verbose

(int) 1 to print information, 0 to not print

trajectory

(list) A list of dictionaries to hold the trajectory information

iteration_number

(int) A running count of which iteration training is on

__init__(verbose=0)

Initialize self. See help(type(self)) for accurate signature.

update_iter()

Updates the iteration number.

This is called by sb_experiment.py to manually increment the iteration number in its training loop.