torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Add num_missing_labels, sum_weights and num_positive_samples

Open iamzainhuda opened this issue 3 weeks ago • 1 comments

Summary: Adds three basic metrics to TorchRec for tracking training statistics:

  1. NumPositiveSamplesMetric - Weighted sum of positive labels
  2. SumWeightsMetric - Total sum of sample weights
  3. NumMissingLabelsMetric - Count of samples with NaN labels

Differential Revision: D87682101

iamzainhuda avatar Nov 26 '25 17:11 iamzainhuda

@iamzainhuda has exported this pull request. If you are a Meta employee, you can view the originating Diff in D87682101.

meta-codesync[bot] avatar Nov 26 '25 17:11 meta-codesync[bot]