Struct StatsTracker 

pub struct StatsTracker { /* private fields */ }

Accumulates per-episode stats and reports summary aggregates.

By default tracks episode_reward (mean) and episode_length (mean). Use the builder methods to add custom stats or change aggregators.

Usage

let mut tracker = StatsTracker::new()
    .with("episode_reward_max", StatSource::TotalReward, Max::default())
    .with_custom("ep_len_last10", |r| r.length as f64, RollingMean::new(10));

// Feed an episode record:
tracker.update(&record);

// Print summary:
let summary = tracker.summary();
println!("mean reward: {:.1}", summary["episode_reward"]);

Implementations

impl StatsTracker

pub fn new() -> Self

Create a tracker with the default stats: episode_reward (mean) and episode_length (mean).

pub fn empty() -> Self

Create a tracker with no default stats.

pub fn with( self, name: impl Into<String>, source: StatSource, aggregator: impl Aggregator + 'static, ) -> Self

Track a predefined stat field with the given aggregator.

pub fn with_custom( self, name: impl Into<String>, f: impl Fn(&EpisodeRecord) -> f64 + Send + Sync + 'static, aggregator: impl Aggregator + 'static, ) -> Self

Track an arbitrary derived value from each episode record.

pub fn update(&mut self, record: &EpisodeRecord)

Feed a completed episode into all tracked stats.

pub fn summary(&self) -> HashMap<String, f64>

Snapshot of all current aggregate values.

pub fn reset(&mut self)

Reset all aggregators (e.g. between eval runs).

Trait Implementations

impl Default for StatsTracker

fn default() -> Self

Returns the “default value” for a type.