StepStatsRenderer

class maze.core.rendering.step_stats_renderer.StepStatsRenderer

Simple statistics rendering based on episode event logs.

Suitable e.g. for ad-hoc plotting of statistics for the current episode during rollout.

static render_stats(episode_event_log: EpisodeEventLog, event_name: str = 'BaseEnvEvents.reward', group_by: str | None = None, aggregation_func: Callable | None = None, metric: str = 'value', cumulative: bool = True)

Queries the event log for the given events, then aggregates them and plots them according to the provided options. By default, a cumulative reward is plotted.

Parameters:

• episode_event_log – The episode event log to draw events from.

• event_name – Name of the event to plot.

• group_by – Attribute of the event that the events should be grouped by when aggregating.

• aggregation_func – Function to aggregate the metric with.

• metric – The metric to plot.

• cumulative – If true, a cumulative sum of the metric is performed (after aggregation).