Metrax Metrics

Metrax provides high-performance, JAX-native evaluation metrics organized into specialized categories. Select a category below to explore the available metrics and their API reference:

• Base
  • metrax.Average
• Audio
  • metrax.SNR
• Classification
  • metrax.AUCPR
  • metrax.AUCROC
  • metrax.Accuracy
  • metrax.BinaryAccuracy
  • metrax.CategoricalAccuracy
  • metrax.FBetaScore
  • metrax.Precision
  • metrax.Recall
  • metrax.SparseCategoricalAccuracy
• Image
  • metrax.CosineSimilarity
  • metrax.Dice
  • metrax.IoU
  • metrax.PSNR
  • metrax.SSIM
• NLP
  • metrax.BLEU
  • metrax.Perplexity
  • metrax.RougeL
  • metrax.RougeN
  • metrax.WER
• Ranking
  • metrax.AveragePrecisionAtK
  • metrax.DCGAtK
  • metrax.MRR
  • metrax.NDCGAtK
  • metrax.PrecisionAtK
  • metrax.RecallAtK
• Regression
  • metrax.MAE
  • metrax.MSE
  • metrax.MSLE
  • metrax.RMSE
  • metrax.RMSLE
  • metrax.RSQUARED
  • metrax.SpearmanRankCorrelation