Struct PredictionAccumulatorState 

pub struct PredictionAccumulatorState { /* private fields */ }

Accumulates raw predictions and targets across batches.

Used by rank-based metrics (AUROC, AUC-PR) that must recompute over the whole epoch. Buffers are freed on reset.

Implementations

impl PredictionAccumulatorState

pub fn new() -> PredictionAccumulatorState

Create a new prediction accumulator state.

pub fn accumulate(&mut self, preds: Tensor<2>, targets: Tensor<2, Bool>)

Accumulate a batch of predictions and targets.

pub fn tensors(&self) -> (Tensor<2>, Tensor<2, Bool>)

All accumulated predictions and targets, concatenated along the samples.

pub fn compute(&mut self, value: f64, format: FormatOptions) -> SerializedEntry

Record the value computed over the accumulated set and return the entry to log.

pub fn value(&self) -> Option<NumericEntry>

Get the current metric value, when available.

pub fn reset(&mut self)

Reset the state, freeing the accumulated tensors.

Trait Implementations

impl Clone for PredictionAccumulatorState

fn clone(&self) -> PredictionAccumulatorState

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Default for PredictionAccumulatorState

fn default() -> PredictionAccumulatorState

Returns the “default value” for a type.