Search — PyTorch-Metrics 1.6.0 documentation

Table of Contents

1.6.0

User Guide

Quick Start
All TorchMetrics
TorchMetrics’ gallery
Audio domain
Image domain
Text domain
Structure Overview
Plotting
Implementing a Metric
TorchMetrics in PyTorch Lightning

Aggregation

Concatenation
Maximum
Mean
Minimum
Running Mean
Running Sum
Sum

Audio

Complex Scale-Invariant Signal-to-Noise Ratio (C-SI-SNR)
Deep Noise Suppression Mean Opinion Score (DNSMOS)
Non-Intrusive Speech Quality Assessment (NISQA v2.0)
Perceptual Evaluation of Speech Quality (PESQ)
Permutation Invariant Training (PIT)
Scale-Invariant Signal-to-Distortion Ratio (SI-SDR)
Scale-Invariant Signal-to-Noise Ratio (SI-SNR)
Short-Time Objective Intelligibility (STOI)
Signal to Distortion Ratio (SDR)
Signal-to-Noise Ratio (SNR)
Source Aggregated Signal-to-Distortion Ratio (SA-SDR)
Speech-to-Reverberation Modulation Energy Ratio (SRMR)

Classification

Accuracy
AUROC
Average Precision
Calibration Error
Cohen Kappa
Confusion Matrix
Coverage Error
Dice
Exact Match
F-1 Score
F-Beta Score
Group Fairness
Hamming Distance
Hinge Loss
Jaccard Index
Label Ranking Average Precision
Label Ranking Loss
Log AUC
Matthews Correlation Coefficient
Negative Predictive Value
Precision
Precision At Fixed Recall
Precision Recall Curve
Recall
Recall At Fixed Precision
ROC
Sensitivity At Specificity
Specificity
Specificity At Sensitivity
Stat Scores

Clustering

Adjusted Mutual Information Score
Adjusted Rand Score
Calinski Harabasz Score
Completeness Score
Davies Bouldin Score
Dunn Index
Fowlkes-Mallows Index
Homogeneity Score
Mutual Information Score
Normalized Mutual Information Score
Rand Score
V-Measure Score

Detection

Complete Intersection Over Union (cIoU)
Distance Intersection Over Union (dIoU)
Generalized Intersection Over Union (gIoU)
Intersection Over Union (IoU)
Mean-Average-Precision (mAP)
Modified Panoptic Quality
Panoptic Quality

Image

Error Relative Global Dim. Synthesis (ERGAS)
Frechet Inception Distance (FID)
Image Gradients
Inception Score
Kernel Inception Distance
Learned Perceptual Image Patch Similarity (LPIPS)
Memorization-Informed Frechet Inception Distance (MiFID)
Multi-Scale SSIM
Peak Signal-to-Noise Ratio (PSNR)
Peak Signal To Noise Ratio With Blocked Effect
Perceptual Path Length (PPL)
Quality with No Reference
Relative Average Spectral Error (RASE)
Root Mean Squared Error Using Sliding Window
Spatial Correlation Coefficient (SCC)
Spatial Distortion Index
Spectral Angle Mapper
Spectral Distortion Index
Structural Similarity Index Measure (SSIM)
Total Variation (TV)
Universal Image Quality Index
Visual Information Fidelity (VIF)

Multimodal

CLIP Image Quality Assessment (CLIP-IQA)
CLIP Score

Nominal

Cramer’s V
Fleiss Kappa
Pearson’s Contingency Coefficient
Theil’s U
Tschuprow’s T

Pairwise

Cosine Similarity
Euclidean Distance
Linear Similarity
Manhattan Distance
Minkowski Distance

Regression

Concordance Corr. Coef.
Cosine Similarity
Critical Success Index (CSI)
Explained Variance
Kendall Rank Corr. Coef.
KL Divergence
Log Cosh Error
Mean Absolute Error (MAE)
Mean Absolute Percentage Error (MAPE)
Mean Squared Error (MSE)
Mean Squared Log Error (MSLE)
Minkowski Distance
Normalized Root Mean Squared Error (NRMSE)
Pearson Corr. Coef.
R2 Score
Relative Squared Error (RSE)
Spearman Corr. Coef.
Symmetric Mean Absolute Percentage Error (SMAPE)
Tweedie Deviance Score
Weighted MAPE

Retrieval

Retrieval AUROC
Retrieval Fall-Out
Retrieval Hit Rate
Retrieval Mean Average Precision (MAP)
Retrieval Mean Reciprocal Rank (MRR)
Retrieval Normalized DCG
Retrieval Precision
Retrieval Precision Recall Curve
Retrieval R-Precision
Retrieval Recall

Segmentation

Dice Score
Generalized Dice Score
Hausdorff Distance
Mean Intersection over Union (mIoU)

Shape

Procrustes Disparity

Text

BERT Score
BLEU Score
Char Error Rate
ChrF Score
Edit Distance
Extended Edit Distance
InfoLM
Match Error Rate
Perplexity
ROUGE Score
Sacre BLEU Score
SQuAD
Translation Edit Rate (TER)
Word Error Rate
Word Info. Lost
Word Info. Preserved

Wrappers

Bootstrapper
Classwise Wrapper
Feature Sharing
Metric Tracker
Min / Max
Multi-output Wrapper
Multi-task Wrapper
Running
Transformations

API Reference

torchmetrics.Metric
torchmetrics.utilities

Community

TorchMetrics Governance
Contributor Covenant Code of Conduct
Contributing
Changelog

Overview
Team management
Production
Security
Open source
- Overview
- PyTorch Lightning
- Fabric
- Lit-GPT
- Torchmetrics
- Litdata
- Lit LLaMA
- Litserve
Examples
Glossary
FAQ

Docs >
Search

Shortcuts

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. Read PyTorch Lightning's Privacy Policy.