torchcell.trainers.neo_regression.ListMLEMetric

class ListMLEMetric(dist_sync_on_step: bool = False)[source]

Bases: Metric

TorchMetric tracking the running mean ListMLE loss over batches.

update(preds: Tensor, target: Tensor) → None[source]: Accumulate the batch-weighted ListMLE loss and sample count.

compute() → Tensor[source]: Return the mean ListMLE loss across all accumulated samples.