Search — PyTorch-Metrics 0.8.2 documentation

Get Started
Blog
Docs
PyTorch Lightning TorchMetrics Lightning Flash Lightning Transformers Lightning Bolts
GitHub
Train on the cloud

Table of Contents

v0.8.2

User Guide

Quick Start
Overview
Implementing a Metric
TorchMetrics in PyTorch Lightning
Using Classification Metrics
Using Retrieval Metrics

Audio

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)

Classification

Accuracy
AUC
AUROC
Average Precision
Binned Average Precision
Binned Precision Recall Curve
Binned Recall At Fixed Precision
Calibration Error
Cohen Kappa
Confusion Matrix
Coverage Error
Dice Score
F1 Score
FBeta Score
Hamming Distance
Hinge Loss
Jaccard Index
KL Divergence
Label Ranking Average Precision
Label Ranking Loss
Matthews Corr. Coef.
Precision
Precision Recall
Precision Recall Curve
Recall
ROC
Specificity
Stat Scores

Image

Error Relative Global Dim. Synthesis (ERGAS)
Frechet Inception Distance (FID)
Image Gradients
Inception Score
Kernel Inception Distance
Learned Perceptual Image Patch Similarity (LPIPS)
Multi-Scale SSIM
Peak Signal-to-Noise Ratio (PSNR)
Spectral Angle Mapper
Spectral Distortion Index
Structural Similarity Index Measure (SSIM)
Universal Image Quality Index

Detection

Mean-Average-Precision (mAP)

Pairwise

Cosine Similarity
Euclidean Distance
Linear Similarity
Manhattan Distance

Regression

Cosine Similarity
Explained Variance
Mean Absolute Error (MAE)
Mean Absolute Percentage Error (MAPE)
Mean Squared Error (MSE)
Mean Squared Log Error (MSLE)
Pearson Corr. Coef.
R2 Score
Spearman Corr. Coef.
Symmetric Mean Absolute Percentage Error (SMAPE)
Tweedie Deviance Score
Weighted MAPE

Retrieval

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

Text

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

Aggregation

Concatenation
Maximum
Mean
Minimum
Sum

Wrappers

Bootstrapper
Classwise Wrapper
Metric Tracker
Min / Max
Multi-output Wrapper

API Reference

torchmetrics.Metric
torchmetrics.utilities.data

Community

TorchMetrics Governance
Contributor Covenant Code of Conduct
Contributing
Changelog

Docs >
Search
Edit on GitHub

Shortcuts

Built with Sphinx using a theme provided by Read the Docs.

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.

Get Started
Blog
Ecosystem

PyTorch Lightning
TorchMetrics
Lightning Flash
Lightning Transformers
Lightning Bolts

Github
Grid.ai