"""训练过程中的通用统计。

分类任务的指标（混淆矩阵、macro-F1 等）在 engine/classification.py 里。
"""

from __future__ import annotations


class AverageMeter:
    def __init__(self) -> None:
        self.total = 0.0
        self.count = 0

    def update(self, value: float, n: int = 1) -> None:
        self.total += float(value) * n
        self.count += n

    @property
    def avg(self) -> float:
        return self.total / self.count if self.count else 0.0