recorders#

Classes#

ExperienceRecorder

Records experience data such as state, action, reward, and next state.

GifRecorder

Captures rgb_array data and creates a gif.

Recorder

class prt_rl.common.recorders.ExperienceRecorder(filename: str)[source]#

Records experience data such as state, action, reward, and next state. This can be used for training or analysis later.

record_experience(experience: dict) None[source]#

Records experience data, such as state, action, reward, and next state. This method can be overridden by subclasses if needed.

record_info(info: dict) None#

Records information from the environment, such as rewards or other metrics. This method can be overridden by subclasses if needed.

class prt_rl.common.recorders.GifRecorder(filename: str, fps: int = 10, loop: bool = True)[source]#

Captures rgb_array data and creates a gif.

Parameters:
  • filename (str) – Filename to save the gif.

  • fps (int) – frames per second

  • loop (bool) – Whether to loop the GIF after it runs. Defaults to True.

close() None[source]#

Saves the captured frames as a GIF.

Parameters:

filename (str) – filename to save GIF to

record_experience(experience: dict) None#

Records experience data, such as state, action, reward, and next state. This method can be overridden by subclasses if needed.

record_info(info: dict) None[source]#

Records information from the environment, such as rewards or other metrics. This method can be overridden by subclasses if needed.

reset()[source]#

Resets the buffer of frames

class prt_rl.common.recorders.Recorder[source]#
record_experience(experience: dict) None[source]#

Records experience data, such as state, action, reward, and next state. This method can be overridden by subclasses if needed.

record_info(info: dict) None[source]#

Records information from the environment, such as rewards or other metrics. This method can be overridden by subclasses if needed.