CHANGELOG.md
CITATION.cff
LICENSE
MANIFEST.in
README.md
requirements.txt
setup.py
requirements/_devel.txt
requirements/_docs.txt
requirements/_doctest.txt
requirements/_integrate.txt
requirements/_tests.txt
requirements/audio.txt
requirements/audio_test.txt
requirements/base.txt
requirements/classification_test.txt
requirements/detection.txt
requirements/detection_test.txt
requirements/image.txt
requirements/image_test.txt
requirements/multimodal.txt
requirements/nominal_test.txt
requirements/segmentation_test.txt
requirements/text.txt
requirements/text_test.txt
requirements/typing.txt
requirements/visual.txt
src/torchmetrics/__about__.py
src/torchmetrics/__init__.py
src/torchmetrics/aggregation.py
src/torchmetrics/collections.py
src/torchmetrics/metric.py
src/torchmetrics/py.typed
src/torchmetrics.egg-info/PKG-INFO
src/torchmetrics.egg-info/SOURCES.txt
src/torchmetrics.egg-info/dependency_links.txt
src/torchmetrics.egg-info/not-zip-safe
src/torchmetrics.egg-info/requires.txt
src/torchmetrics.egg-info/top_level.txt
src/torchmetrics/audio/__init__.py
src/torchmetrics/audio/_deprecated.py
src/torchmetrics/audio/pesq.py
src/torchmetrics/audio/pit.py
src/torchmetrics/audio/sdr.py
src/torchmetrics/audio/snr.py
src/torchmetrics/audio/srmr.py
src/torchmetrics/audio/stoi.py
src/torchmetrics/classification/__init__.py
src/torchmetrics/classification/accuracy.py
src/torchmetrics/classification/auroc.py
src/torchmetrics/classification/average_precision.py
src/torchmetrics/classification/base.py
src/torchmetrics/classification/calibration_error.py
src/torchmetrics/classification/cohen_kappa.py
src/torchmetrics/classification/confusion_matrix.py
src/torchmetrics/classification/dice.py
src/torchmetrics/classification/exact_match.py
src/torchmetrics/classification/f_beta.py
src/torchmetrics/classification/group_fairness.py
src/torchmetrics/classification/hamming.py
src/torchmetrics/classification/hinge.py
src/torchmetrics/classification/jaccard.py
src/torchmetrics/classification/matthews_corrcoef.py
src/torchmetrics/classification/precision_fixed_recall.py
src/torchmetrics/classification/precision_recall.py
src/torchmetrics/classification/precision_recall_curve.py
src/torchmetrics/classification/ranking.py
src/torchmetrics/classification/recall_fixed_precision.py
src/torchmetrics/classification/roc.py
src/torchmetrics/classification/specificity.py
src/torchmetrics/classification/specificity_sensitivity.py
src/torchmetrics/classification/stat_scores.py
src/torchmetrics/clustering/__init__.py
src/torchmetrics/clustering/adjusted_mutual_info_score.py
src/torchmetrics/clustering/adjusted_rand_score.py
src/torchmetrics/clustering/calinski_harabasz_score.py
src/torchmetrics/clustering/davies_bouldin_score.py
src/torchmetrics/clustering/dunn_index.py
src/torchmetrics/clustering/fowlkes_mallows_index.py
src/torchmetrics/clustering/homogeneity_completeness_v_measure.py
src/torchmetrics/clustering/mutual_info_score.py
src/torchmetrics/clustering/normalized_mutual_info_score.py
src/torchmetrics/clustering/rand_score.py
src/torchmetrics/detection/__init__.py
src/torchmetrics/detection/_deprecated.py
src/torchmetrics/detection/_mean_ap.py
src/torchmetrics/detection/ciou.py
src/torchmetrics/detection/diou.py
src/torchmetrics/detection/giou.py
src/torchmetrics/detection/helpers.py
src/torchmetrics/detection/iou.py
src/torchmetrics/detection/mean_ap.py
src/torchmetrics/detection/panoptic_qualities.py
src/torchmetrics/functional/__init__.py
src/torchmetrics/functional/audio/__init__.py
src/torchmetrics/functional/audio/_deprecated.py
src/torchmetrics/functional/audio/pesq.py
src/torchmetrics/functional/audio/pit.py
src/torchmetrics/functional/audio/sdr.py
src/torchmetrics/functional/audio/snr.py
src/torchmetrics/functional/audio/srmr.py
src/torchmetrics/functional/audio/stoi.py
src/torchmetrics/functional/classification/__init__.py
src/torchmetrics/functional/classification/accuracy.py
src/torchmetrics/functional/classification/auroc.py
src/torchmetrics/functional/classification/average_precision.py
src/torchmetrics/functional/classification/calibration_error.py
src/torchmetrics/functional/classification/cohen_kappa.py
src/torchmetrics/functional/classification/confusion_matrix.py
src/torchmetrics/functional/classification/dice.py
src/torchmetrics/functional/classification/exact_match.py
src/torchmetrics/functional/classification/f_beta.py
src/torchmetrics/functional/classification/group_fairness.py
src/torchmetrics/functional/classification/hamming.py
src/torchmetrics/functional/classification/hinge.py
src/torchmetrics/functional/classification/jaccard.py
src/torchmetrics/functional/classification/matthews_corrcoef.py
src/torchmetrics/functional/classification/precision_fixed_recall.py
src/torchmetrics/functional/classification/precision_recall.py
src/torchmetrics/functional/classification/precision_recall_curve.py
src/torchmetrics/functional/classification/ranking.py
src/torchmetrics/functional/classification/recall_fixed_precision.py
src/torchmetrics/functional/classification/roc.py
src/torchmetrics/functional/classification/specificity.py
src/torchmetrics/functional/classification/specificity_sensitivity.py
src/torchmetrics/functional/classification/stat_scores.py
src/torchmetrics/functional/clustering/__init__.py
src/torchmetrics/functional/clustering/adjusted_mutual_info_score.py
src/torchmetrics/functional/clustering/adjusted_rand_score.py
src/torchmetrics/functional/clustering/calinski_harabasz_score.py
src/torchmetrics/functional/clustering/davies_bouldin_score.py
src/torchmetrics/functional/clustering/dunn_index.py
src/torchmetrics/functional/clustering/fowlkes_mallows_index.py
src/torchmetrics/functional/clustering/homogeneity_completeness_v_measure.py
src/torchmetrics/functional/clustering/mutual_info_score.py
src/torchmetrics/functional/clustering/normalized_mutual_info_score.py
src/torchmetrics/functional/clustering/rand_score.py
src/torchmetrics/functional/clustering/utils.py
src/torchmetrics/functional/detection/__init__.py
src/torchmetrics/functional/detection/_deprecated.py
src/torchmetrics/functional/detection/_panoptic_quality_common.py
src/torchmetrics/functional/detection/ciou.py
src/torchmetrics/functional/detection/diou.py
src/torchmetrics/functional/detection/giou.py
src/torchmetrics/functional/detection/iou.py
src/torchmetrics/functional/detection/panoptic_qualities.py
src/torchmetrics/functional/image/__init__.py
src/torchmetrics/functional/image/_deprecated.py
src/torchmetrics/functional/image/d_lambda.py
src/torchmetrics/functional/image/d_s.py
src/torchmetrics/functional/image/ergas.py
src/torchmetrics/functional/image/gradients.py
src/torchmetrics/functional/image/helper.py
src/torchmetrics/functional/image/lpips.py
src/torchmetrics/functional/image/perceptual_path_length.py
src/torchmetrics/functional/image/psnr.py
src/torchmetrics/functional/image/psnrb.py
src/torchmetrics/functional/image/rase.py
src/torchmetrics/functional/image/rmse_sw.py
src/torchmetrics/functional/image/sam.py
src/torchmetrics/functional/image/scc.py
src/torchmetrics/functional/image/ssim.py
src/torchmetrics/functional/image/tv.py
src/torchmetrics/functional/image/uqi.py
src/torchmetrics/functional/image/vif.py
src/torchmetrics/functional/image/lpips_models/alex.pth
src/torchmetrics/functional/image/lpips_models/squeeze.pth
src/torchmetrics/functional/image/lpips_models/vgg.pth
src/torchmetrics/functional/multimodal/__init__.py
src/torchmetrics/functional/multimodal/clip_iqa.py
src/torchmetrics/functional/multimodal/clip_score.py
src/torchmetrics/functional/nominal/__init__.py
src/torchmetrics/functional/nominal/cramers.py
src/torchmetrics/functional/nominal/fleiss_kappa.py
src/torchmetrics/functional/nominal/pearson.py
src/torchmetrics/functional/nominal/theils_u.py
src/torchmetrics/functional/nominal/tschuprows.py
src/torchmetrics/functional/nominal/utils.py
src/torchmetrics/functional/pairwise/__init__.py
src/torchmetrics/functional/pairwise/cosine.py
src/torchmetrics/functional/pairwise/euclidean.py
src/torchmetrics/functional/pairwise/helpers.py
src/torchmetrics/functional/pairwise/linear.py
src/torchmetrics/functional/pairwise/manhattan.py
src/torchmetrics/functional/pairwise/minkowski.py
src/torchmetrics/functional/regression/__init__.py
src/torchmetrics/functional/regression/concordance.py
src/torchmetrics/functional/regression/cosine_similarity.py
src/torchmetrics/functional/regression/csi.py
src/torchmetrics/functional/regression/explained_variance.py
src/torchmetrics/functional/regression/kendall.py
src/torchmetrics/functional/regression/kl_divergence.py
src/torchmetrics/functional/regression/log_cosh.py
src/torchmetrics/functional/regression/log_mse.py
src/torchmetrics/functional/regression/mae.py
src/torchmetrics/functional/regression/mape.py
src/torchmetrics/functional/regression/minkowski.py
src/torchmetrics/functional/regression/mse.py
src/torchmetrics/functional/regression/pearson.py
src/torchmetrics/functional/regression/r2.py
src/torchmetrics/functional/regression/rse.py
src/torchmetrics/functional/regression/spearman.py
src/torchmetrics/functional/regression/symmetric_mape.py
src/torchmetrics/functional/regression/tweedie_deviance.py
src/torchmetrics/functional/regression/utils.py
src/torchmetrics/functional/regression/wmape.py
src/torchmetrics/functional/retrieval/__init__.py
src/torchmetrics/functional/retrieval/_deprecated.py
src/torchmetrics/functional/retrieval/auroc.py
src/torchmetrics/functional/retrieval/average_precision.py
src/torchmetrics/functional/retrieval/fall_out.py
src/torchmetrics/functional/retrieval/hit_rate.py
src/torchmetrics/functional/retrieval/ndcg.py
src/torchmetrics/functional/retrieval/precision.py
src/torchmetrics/functional/retrieval/precision_recall_curve.py
src/torchmetrics/functional/retrieval/r_precision.py
src/torchmetrics/functional/retrieval/recall.py
src/torchmetrics/functional/retrieval/reciprocal_rank.py
src/torchmetrics/functional/segmentation/__init__.py
src/torchmetrics/functional/segmentation/utils.py
src/torchmetrics/functional/text/__init__.py
src/torchmetrics/functional/text/_deprecated.py
src/torchmetrics/functional/text/bert.py
src/torchmetrics/functional/text/bleu.py
src/torchmetrics/functional/text/cer.py
src/torchmetrics/functional/text/chrf.py
src/torchmetrics/functional/text/edit.py
src/torchmetrics/functional/text/eed.py
src/torchmetrics/functional/text/helper.py
src/torchmetrics/functional/text/helper_embedding_metric.py
src/torchmetrics/functional/text/infolm.py
src/torchmetrics/functional/text/mer.py
src/torchmetrics/functional/text/perplexity.py
src/torchmetrics/functional/text/rouge.py
src/torchmetrics/functional/text/sacre_bleu.py
src/torchmetrics/functional/text/squad.py
src/torchmetrics/functional/text/ter.py
src/torchmetrics/functional/text/wer.py
src/torchmetrics/functional/text/wil.py
src/torchmetrics/functional/text/wip.py
src/torchmetrics/image/__init__.py
src/torchmetrics/image/_deprecated.py
src/torchmetrics/image/d_lambda.py
src/torchmetrics/image/d_s.py
src/torchmetrics/image/ergas.py
src/torchmetrics/image/fid.py
src/torchmetrics/image/inception.py
src/torchmetrics/image/kid.py
src/torchmetrics/image/lpip.py
src/torchmetrics/image/mifid.py
src/torchmetrics/image/perceptual_path_length.py
src/torchmetrics/image/psnr.py
src/torchmetrics/image/psnrb.py
src/torchmetrics/image/rase.py
src/torchmetrics/image/rmse_sw.py
src/torchmetrics/image/sam.py
src/torchmetrics/image/scc.py
src/torchmetrics/image/ssim.py
src/torchmetrics/image/tv.py
src/torchmetrics/image/uqi.py
src/torchmetrics/image/vif.py
src/torchmetrics/multimodal/__init__.py
src/torchmetrics/multimodal/clip_iqa.py
src/torchmetrics/multimodal/clip_score.py
src/torchmetrics/nominal/__init__.py
src/torchmetrics/nominal/cramers.py
src/torchmetrics/nominal/fleiss_kappa.py
src/torchmetrics/nominal/pearson.py
src/torchmetrics/nominal/theils_u.py
src/torchmetrics/nominal/tschuprows.py
src/torchmetrics/regression/__init__.py
src/torchmetrics/regression/concordance.py
src/torchmetrics/regression/cosine_similarity.py
src/torchmetrics/regression/csi.py
src/torchmetrics/regression/explained_variance.py
src/torchmetrics/regression/kendall.py
src/torchmetrics/regression/kl_divergence.py
src/torchmetrics/regression/log_cosh.py
src/torchmetrics/regression/log_mse.py
src/torchmetrics/regression/mae.py
src/torchmetrics/regression/mape.py
src/torchmetrics/regression/minkowski.py
src/torchmetrics/regression/mse.py
src/torchmetrics/regression/pearson.py
src/torchmetrics/regression/r2.py
src/torchmetrics/regression/rse.py
src/torchmetrics/regression/spearman.py
src/torchmetrics/regression/symmetric_mape.py
src/torchmetrics/regression/tweedie_deviance.py
src/torchmetrics/regression/wmape.py
src/torchmetrics/retrieval/__init__.py
src/torchmetrics/retrieval/_deprecated.py
src/torchmetrics/retrieval/auroc.py
src/torchmetrics/retrieval/average_precision.py
src/torchmetrics/retrieval/base.py
src/torchmetrics/retrieval/fall_out.py
src/torchmetrics/retrieval/hit_rate.py
src/torchmetrics/retrieval/ndcg.py
src/torchmetrics/retrieval/precision.py
src/torchmetrics/retrieval/precision_recall_curve.py
src/torchmetrics/retrieval/r_precision.py
src/torchmetrics/retrieval/recall.py
src/torchmetrics/retrieval/reciprocal_rank.py
src/torchmetrics/text/__init__.py
src/torchmetrics/text/_deprecated.py
src/torchmetrics/text/bert.py
src/torchmetrics/text/bleu.py
src/torchmetrics/text/cer.py
src/torchmetrics/text/chrf.py
src/torchmetrics/text/edit.py
src/torchmetrics/text/eed.py
src/torchmetrics/text/infolm.py
src/torchmetrics/text/mer.py
src/torchmetrics/text/perplexity.py
src/torchmetrics/text/rouge.py
src/torchmetrics/text/sacre_bleu.py
src/torchmetrics/text/squad.py
src/torchmetrics/text/ter.py
src/torchmetrics/text/wer.py
src/torchmetrics/text/wil.py
src/torchmetrics/text/wip.py
src/torchmetrics/utilities/__init__.py
src/torchmetrics/utilities/checks.py
src/torchmetrics/utilities/compute.py
src/torchmetrics/utilities/data.py
src/torchmetrics/utilities/distributed.py
src/torchmetrics/utilities/enums.py
src/torchmetrics/utilities/exceptions.py
src/torchmetrics/utilities/imports.py
src/torchmetrics/utilities/plot.py
src/torchmetrics/utilities/prints.py
src/torchmetrics/wrappers/__init__.py
src/torchmetrics/wrappers/abstract.py
src/torchmetrics/wrappers/bootstrapping.py
src/torchmetrics/wrappers/classwise.py
src/torchmetrics/wrappers/feature_share.py
src/torchmetrics/wrappers/minmax.py
src/torchmetrics/wrappers/multioutput.py
src/torchmetrics/wrappers/multitask.py
src/torchmetrics/wrappers/running.py
src/torchmetrics/wrappers/tracker.py