├── .azure ├── gpu-integrations.yml ├── gpu-nuke-cache.yml └── gpu-unittests.yml ├── .codecov.yml ├── .devcontainer ├── Dockerfile └── devcontainer.json ├── .github ├── CODEOWNERS ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── ISSUE_TEMPLATE │ ├── bug_report.md │ ├── config.yml │ ├── documentation.md │ └── feature_request.md ├── PULL_REQUEST_TEMPLATE.md ├── actions │ ├── pull-caches │ │ └── action.yml │ └── push-caches │ │ └── action.yml ├── assistant.py ├── dependabot.yml ├── labeling-config.yml ├── markdown-links-config.json ├── mergify.yml ├── stale.yml └── workflows │ ├── _focus-diff.yml │ ├── ci-checks.yml │ ├── ci-integrate.yml │ ├── ci-rtfd.yml │ ├── ci-tests.yml │ ├── clear-cache.yml │ ├── cmd-help.yml │ ├── cmd-rebase.yml │ ├── docker-build.yml │ ├── docs-build.yml │ ├── greetings.yml │ ├── labeler.yml │ ├── publish-pkg.yml │ └── slash-cmd-dispatch.yml ├── .gitignore ├── .lightning └── workflows │ ├── integrations.yml │ └── uinttests.yml ├── .pre-commit-config.yaml ├── CHANGELOG.md ├── CITATION.cff ├── LICENSE ├── MANIFEST.in ├── Makefile ├── README.md ├── _samples ├── bert_score-own_model.py ├── detection_map.py ├── plotting.py └── rouge_score-own_normalizer_and_tokenizer.py ├── dockers ├── README.md └── ubuntu-cuda │ └── Dockerfile ├── docs ├── .build_docs.sh ├── .readthedocs.yaml ├── Makefile ├── README.md ├── make.bat ├── paper_JOSS │ ├── paper.bib │ └── paper.md ├── redirect.html ├── rtfd-build.sh └── source │ ├── _static │ ├── copybutton.js │ ├── css │ │ └── custom.css │ └── images │ │ ├── icon.png │ │ ├── icon.svg │ │ ├── logo.png │ │ ├── logo.svg │ │ ├── logo_light.svg │ │ └── plot_example.png │ ├── _templates │ ├── autosummary │ │ └── module.rst │ ├── classtemplate.rst │ ├── layout.html │ └── theme_variables.jinja │ ├── aggregation │ ├── cat.rst │ ├── max.rst │ ├── mean.rst │ ├── min.rst │ ├── running_mean.rst │ ├── running_sum.rst │ └── sum.rst │ ├── all-metrics.rst │ ├── audio │ ├── complex_scale_invariant_signal_noise_ratio.rst │ ├── deep_noise_suppression_mean_opinion_score.rst │ ├── non_intrusive_speech_quality_assessment.rst │ ├── perceptual_evaluation_speech_quality.rst │ ├── permutation_invariant_training.rst │ ├── scale_invariant_signal_distortion_ratio.rst │ ├── scale_invariant_signal_noise_ratio.rst │ ├── short_time_objective_intelligibility.rst │ ├── signal_distortion_ratio.rst │ ├── signal_noise_ratio.rst │ ├── source_aggregated_signal_distortion_ratio.rst │ └── speech_reverberation_modulation_energy_ratio.rst │ ├── classification │ ├── accuracy.rst │ ├── auroc.rst │ ├── average_precision.rst │ ├── calibration_error.rst │ ├── cohen_kappa.rst │ ├── confusion_matrix.rst │ ├── coverage_error.rst │ ├── eer.rst │ ├── exact_match.rst │ ├── f1_score.rst │ ├── fbeta_score.rst │ ├── group_fairness.rst │ ├── hamming_distance.rst │ ├── hinge_loss.rst │ ├── jaccard_index.rst │ ├── label_ranking_average_precision.rst │ ├── label_ranking_loss.rst │ ├── logauc.rst │ ├── matthews_corr_coef.rst │ ├── negative_predictive_value.rst │ ├── precision.rst │ ├── precision_at_fixed_recall.rst │ ├── precision_recall_curve.rst │ ├── recall.rst │ ├── recall_at_fixed_precision.rst │ ├── roc.rst │ ├── sensitivity_at_specificity.rst │ ├── specificity.rst │ ├── specificity_at_sensitivity.rst │ └── stat_scores.rst │ ├── clustering │ ├── adjusted_mutual_info_score.rst │ ├── adjusted_rand_score.rst │ ├── calinski_harabasz_score.rst │ ├── cluster_accuracy.rst │ ├── completeness_score.rst │ ├── davies_bouldin_score.rst │ ├── dunn_index.rst │ ├── fowlkes_mallows_index.rst │ ├── homogeneity_score.rst │ ├── mutual_info_score.rst │ ├── normalized_mutual_info_score.rst │ ├── rand_score.rst │ └── v_measure_score.rst │ ├── conf.py │ ├── detection │ ├── complete_intersection_over_union.rst │ ├── distance_intersection_over_union.rst │ ├── generalized_intersection_over_union.rst │ ├── intersection_over_union.rst │ ├── mean_average_precision.rst │ ├── modified_panoptic_quality.rst │ └── panoptic_quality.rst │ ├── governance.rst │ ├── image │ ├── arniqa.rst │ ├── dists.rst │ ├── error_relative_global_dimensionless_synthesis.rst │ ├── frechet_inception_distance.rst │ ├── image_gradients.rst │ ├── inception_score.rst │ ├── kernel_inception_distance.rst │ ├── learned_perceptual_image_patch_similarity.rst │ ├── mifid.rst │ ├── multi_scale_structural_similarity.rst │ ├── peak_signal_noise_ratio.rst │ ├── peak_signal_to_noise_with_block.rst │ ├── perceptual_path_length.rst │ ├── quality_with_no_reference.rst │ ├── relative_average_spectral_error.rst │ ├── root_mean_squared_error_using_sliding_window.rst │ ├── spatial_correlation_coefficient.rst │ ├── spatial_distortion_index.rst │ ├── spectral_angle_mapper.rst │ ├── spectral_distortion_index.rst │ ├── structural_similarity.rst │ ├── total_variation.rst │ ├── universal_image_quality_index.rst │ └── visual_information_fidelity.rst │ ├── index.rst │ ├── links.rst │ ├── multimodal │ ├── clip_iqa.rst │ ├── clip_score.rst │ └── lve.rst │ ├── nominal │ ├── cramers_v.rst │ ├── fleiss_kappa.rst │ ├── pearsons_contingency_coefficient.rst │ ├── theils_u.rst │ └── tschuprows_t.rst │ ├── pages │ ├── implement.rst │ ├── lightning.rst │ ├── overview.rst │ ├── plotting.rst │ └── quickstart.rst │ ├── pairwise │ ├── cosine_similarity.rst │ ├── euclidean_distance.rst │ ├── linear_similarity.rst │ ├── manhattan_distance.rst │ └── minkowski_distance.rst │ ├── pyplots │ ├── binary_accuracy.py │ ├── binary_accuracy_multistep.py │ ├── collection_binary.py │ ├── collection_binary_together.py │ ├── confusion_matrix.py │ ├── multiclass_accuracy.py │ └── tracker_binary.py │ ├── references │ ├── metric.rst │ └── utilities.rst │ ├── regression │ ├── concordance_corr_coef.rst │ ├── cosine_similarity.rst │ ├── critical_success_index.rst │ ├── crps.rst │ ├── explained_variance.rst │ ├── js_divergence.rst │ ├── kendall_rank_corr_coef.rst │ ├── kl_divergence.rst │ ├── log_cosh_error.rst │ ├── mean_absolute_error.rst │ ├── mean_absolute_percentage_error.rst │ ├── mean_squared_error.rst │ ├── mean_squared_log_error.rst │ ├── minkowski_distance.rst │ ├── normalized_root_mean_squared_error.rst │ ├── pearson_corr_coef.rst │ ├── r2_score.rst │ ├── rse.rst │ ├── spearman_corr_coef.rst │ ├── symmetric_mean_absolute_percentage_error.rst │ ├── tweedie_deviance_score.rst │ └── weighted_mean_absolute_percentage_error.rst │ ├── retrieval │ ├── auroc.rst │ ├── fall_out.rst │ ├── hit_rate.rst │ ├── map.rst │ ├── mrr.rst │ ├── normalized_dcg.rst │ ├── precision.rst │ ├── precision_recall_curve.rst │ ├── r_precision.rst │ └── recall.rst │ ├── segmentation │ ├── dice.rst │ ├── generalized_dice.rst │ ├── hausdorff_distance.rst │ └── mean_iou.rst │ ├── shape │ └── procrustes.rst │ ├── text │ ├── bert_score.rst │ ├── bleu_score.rst │ ├── char_error_rate.rst │ ├── chrf_score.rst │ ├── edit.rst │ ├── extended_edit_distance.rst │ ├── infolm.rst │ ├── match_error_rate.rst │ ├── perplexity.rst │ ├── rouge_score.rst │ ├── sacre_bleu_score.rst │ ├── squad.rst │ ├── translation_edit_rate.rst │ ├── word_error_rate.rst │ ├── word_info_lost.rst │ └── word_info_preserved.rst │ ├── video │ └── vmaf.rst │ └── wrappers │ ├── bootstrapper.rst │ ├── classwise_wrapper.rst │ ├── feature_share.rst │ ├── metric_tracker.rst │ ├── min_max.rst │ ├── multi_output_wrapper.rst │ ├── multi_task_wrapper.rst │ ├── running.rst │ └── transformations.rst ├── examples ├── README.rst ├── audio │ ├── README.rst │ ├── pesq.py │ ├── signal_to_noise_ratio.py │ └── text_to_speech.py ├── image │ ├── README.rst │ ├── clip_score.py │ └── spatial_correlation_coef.py └── text │ ├── README.rst │ ├── bertscore.py │ ├── perplexity.py │ └── rouge.py ├── pyproject.toml ├── requirements.txt ├── requirements ├── README.md ├── _ci.txt ├── _devel.txt ├── _docs.txt ├── _doctest.txt ├── _integrate.txt ├── _tests.txt ├── audio.txt ├── audio_test.txt ├── base.txt ├── classification_test.txt ├── clustering.txt ├── clustering_test.txt ├── debug.txt ├── detection.txt ├── detection_test.txt ├── image.txt ├── image_test.txt ├── integrate.txt ├── multimodal.txt ├── nominal_test.txt ├── regression_test.txt ├── segmentation_test.txt ├── text.txt ├── text_test.txt ├── typing.txt ├── video.txt └── visual.txt ├── setup.py ├── src ├── conftest.py └── torchmetrics │ ├── __about__.py │ ├── __init__.py │ ├── aggregation.py │ ├── audio │ ├── __init__.py │ ├── _deprecated.py │ ├── dnsmos.py │ ├── nisqa.py │ ├── pesq.py │ ├── pit.py │ ├── sdr.py │ ├── snr.py │ ├── srmr.py │ └── stoi.py │ ├── classification │ ├── __init__.py │ ├── accuracy.py │ ├── auroc.py │ ├── average_precision.py │ ├── base.py │ ├── calibration_error.py │ ├── cohen_kappa.py │ ├── confusion_matrix.py │ ├── eer.py │ ├── exact_match.py │ ├── f_beta.py │ ├── group_fairness.py │ ├── hamming.py │ ├── hinge.py │ ├── jaccard.py │ ├── logauc.py │ ├── matthews_corrcoef.py │ ├── negative_predictive_value.py │ ├── precision_fixed_recall.py │ ├── precision_recall.py │ ├── precision_recall_curve.py │ ├── ranking.py │ ├── recall_fixed_precision.py │ ├── roc.py │ ├── sensitivity_specificity.py │ ├── specificity.py │ ├── specificity_sensitivity.py │ └── stat_scores.py │ ├── clustering │ ├── __init__.py │ ├── adjusted_mutual_info_score.py │ ├── adjusted_rand_score.py │ ├── calinski_harabasz_score.py │ ├── cluster_accuracy.py │ ├── davies_bouldin_score.py │ ├── dunn_index.py │ ├── fowlkes_mallows_index.py │ ├── homogeneity_completeness_v_measure.py │ ├── mutual_info_score.py │ ├── normalized_mutual_info_score.py │ └── rand_score.py │ ├── collections.py │ ├── detection │ ├── __init__.py │ ├── _deprecated.py │ ├── _mean_ap.py │ ├── ciou.py │ ├── diou.py │ ├── giou.py │ ├── helpers.py │ ├── iou.py │ ├── mean_ap.py │ └── panoptic_qualities.py │ ├── functional │ ├── __init__.py │ ├── audio │ │ ├── __init__.py │ │ ├── _deprecated.py │ │ ├── dnsmos.py │ │ ├── nisqa.py │ │ ├── pesq.py │ │ ├── pit.py │ │ ├── sdr.py │ │ ├── snr.py │ │ ├── srmr.py │ │ └── stoi.py │ ├── classification │ │ ├── __init__.py │ │ ├── accuracy.py │ │ ├── auroc.py │ │ ├── average_precision.py │ │ ├── calibration_error.py │ │ ├── cohen_kappa.py │ │ ├── confusion_matrix.py │ │ ├── eer.py │ │ ├── exact_match.py │ │ ├── f_beta.py │ │ ├── group_fairness.py │ │ ├── hamming.py │ │ ├── hinge.py │ │ ├── jaccard.py │ │ ├── logauc.py │ │ ├── matthews_corrcoef.py │ │ ├── negative_predictive_value.py │ │ ├── precision_fixed_recall.py │ │ ├── precision_recall.py │ │ ├── precision_recall_curve.py │ │ ├── ranking.py │ │ ├── recall_fixed_precision.py │ │ ├── roc.py │ │ ├── sensitivity_specificity.py │ │ ├── specificity.py │ │ ├── specificity_sensitivity.py │ │ └── stat_scores.py │ ├── clustering │ │ ├── __init__.py │ │ ├── adjusted_mutual_info_score.py │ │ ├── adjusted_rand_score.py │ │ ├── calinski_harabasz_score.py │ │ ├── cluster_accuracy.py │ │ ├── davies_bouldin_score.py │ │ ├── dunn_index.py │ │ ├── fowlkes_mallows_index.py │ │ ├── homogeneity_completeness_v_measure.py │ │ ├── mutual_info_score.py │ │ ├── normalized_mutual_info_score.py │ │ ├── rand_score.py │ │ └── utils.py │ ├── detection │ │ ├── __init__.py │ │ ├── _deprecated.py │ │ ├── _panoptic_quality_common.py │ │ ├── ciou.py │ │ ├── diou.py │ │ ├── giou.py │ │ ├── iou.py │ │ ├── map.py │ │ └── panoptic_qualities.py │ ├── image │ │ ├── __init__.py │ │ ├── _deprecated.py │ │ ├── arniqa.py │ │ ├── d_lambda.py │ │ ├── d_s.py │ │ ├── dists.py │ │ ├── dists_models │ │ │ └── weights.pt │ │ ├── ergas.py │ │ ├── gradients.py │ │ ├── lpips.py │ │ ├── lpips_models │ │ │ ├── alex.pth │ │ │ ├── squeeze.pth │ │ │ └── vgg.pth │ │ ├── perceptual_path_length.py │ │ ├── psnr.py │ │ ├── psnrb.py │ │ ├── qnr.py │ │ ├── rase.py │ │ ├── rmse_sw.py │ │ ├── sam.py │ │ ├── scc.py │ │ ├── ssim.py │ │ ├── tv.py │ │ ├── uqi.py │ │ ├── utils.py │ │ └── vif.py │ ├── multimodal │ │ ├── __init__.py │ │ ├── clip_iqa.py │ │ ├── clip_score.py │ │ └── lve.py │ ├── nominal │ │ ├── __init__.py │ │ ├── cramers.py │ │ ├── fleiss_kappa.py │ │ ├── pearson.py │ │ ├── theils_u.py │ │ ├── tschuprows.py │ │ └── utils.py │ ├── pairwise │ │ ├── __init__.py │ │ ├── cosine.py │ │ ├── euclidean.py │ │ ├── helpers.py │ │ ├── linear.py │ │ ├── manhattan.py │ │ └── minkowski.py │ ├── regression │ │ ├── __init__.py │ │ ├── concordance.py │ │ ├── cosine_similarity.py │ │ ├── crps.py │ │ ├── csi.py │ │ ├── explained_variance.py │ │ ├── js_divergence.py │ │ ├── kendall.py │ │ ├── kl_divergence.py │ │ ├── log_cosh.py │ │ ├── log_mse.py │ │ ├── mae.py │ │ ├── mape.py │ │ ├── minkowski.py │ │ ├── mse.py │ │ ├── nrmse.py │ │ ├── pearson.py │ │ ├── r2.py │ │ ├── rse.py │ │ ├── spearman.py │ │ ├── symmetric_mape.py │ │ ├── tweedie_deviance.py │ │ ├── utils.py │ │ └── wmape.py │ ├── retrieval │ │ ├── __init__.py │ │ ├── _deprecated.py │ │ ├── auroc.py │ │ ├── average_precision.py │ │ ├── fall_out.py │ │ ├── hit_rate.py │ │ ├── ndcg.py │ │ ├── precision.py │ │ ├── precision_recall_curve.py │ │ ├── r_precision.py │ │ ├── recall.py │ │ └── reciprocal_rank.py │ ├── segmentation │ │ ├── __init__.py │ │ ├── dice.py │ │ ├── generalized_dice.py │ │ ├── hausdorff_distance.py │ │ ├── mean_iou.py │ │ └── utils.py │ ├── shape │ │ ├── __init__.py │ │ └── procrustes.py │ ├── text │ │ ├── __init__.py │ │ ├── _deprecated.py │ │ ├── bert.py │ │ ├── bleu.py │ │ ├── cer.py │ │ ├── chrf.py │ │ ├── edit.py │ │ ├── eed.py │ │ ├── helper.py │ │ ├── helper_embedding_metric.py │ │ ├── infolm.py │ │ ├── mer.py │ │ ├── perplexity.py │ │ ├── rouge.py │ │ ├── sacre_bleu.py │ │ ├── squad.py │ │ ├── ter.py │ │ ├── wer.py │ │ ├── wil.py │ │ └── wip.py │ └── video │ │ ├── __init__.py │ │ └── vmaf.py │ ├── image │ ├── __init__.py │ ├── _deprecated.py │ ├── arniqa.py │ ├── d_lambda.py │ ├── d_s.py │ ├── dists.py │ ├── ergas.py │ ├── fid.py │ ├── inception.py │ ├── kid.py │ ├── lpip.py │ ├── mifid.py │ ├── perceptual_path_length.py │ ├── psnr.py │ ├── psnrb.py │ ├── qnr.py │ ├── rase.py │ ├── rmse_sw.py │ ├── sam.py │ ├── scc.py │ ├── ssim.py │ ├── tv.py │ ├── uqi.py │ └── vif.py │ ├── metric.py │ ├── multimodal │ ├── __init__.py │ ├── clip_iqa.py │ ├── clip_score.py │ └── lve.py │ ├── nominal │ ├── __init__.py │ ├── cramers.py │ ├── fleiss_kappa.py │ ├── pearson.py │ ├── theils_u.py │ └── tschuprows.py │ ├── py.typed │ ├── regression │ ├── __init__.py │ ├── concordance.py │ ├── cosine_similarity.py │ ├── crps.py │ ├── csi.py │ ├── explained_variance.py │ ├── js_divergence.py │ ├── kendall.py │ ├── kl_divergence.py │ ├── log_cosh.py │ ├── log_mse.py │ ├── mae.py │ ├── mape.py │ ├── minkowski.py │ ├── mse.py │ ├── nrmse.py │ ├── pearson.py │ ├── r2.py │ ├── rse.py │ ├── spearman.py │ ├── symmetric_mape.py │ ├── tweedie_deviance.py │ └── wmape.py │ ├── retrieval │ ├── __init__.py │ ├── _deprecated.py │ ├── auroc.py │ ├── average_precision.py │ ├── base.py │ ├── fall_out.py │ ├── hit_rate.py │ ├── ndcg.py │ ├── precision.py │ ├── precision_recall_curve.py │ ├── r_precision.py │ ├── recall.py │ └── reciprocal_rank.py │ ├── segmentation │ ├── __init__.py │ ├── dice.py │ ├── generalized_dice.py │ ├── hausdorff_distance.py │ └── mean_iou.py │ ├── shape │ ├── __init__.py │ └── procrustes.py │ ├── text │ ├── __init__.py │ ├── _deprecated.py │ ├── bert.py │ ├── bleu.py │ ├── cer.py │ ├── chrf.py │ ├── edit.py │ ├── eed.py │ ├── infolm.py │ ├── mer.py │ ├── perplexity.py │ ├── rouge.py │ ├── sacre_bleu.py │ ├── squad.py │ ├── ter.py │ ├── wer.py │ ├── wil.py │ └── wip.py │ ├── utilities │ ├── __init__.py │ ├── checks.py │ ├── compute.py │ ├── data.py │ ├── distributed.py │ ├── enums.py │ ├── exceptions.py │ ├── imports.py │ ├── plot.py │ └── prints.py │ ├── video │ ├── __init__.py │ └── vmaf.py │ └── wrappers │ ├── __init__.py │ ├── abstract.py │ ├── bootstrapping.py │ ├── classwise.py │ ├── feature_share.py │ ├── minmax.py │ ├── multioutput.py │ ├── multitask.py │ ├── running.py │ ├── tracker.py │ └── transformations.py └── tests ├── README.md ├── integrations ├── __init__.py ├── conftest.py ├── lightning │ ├── __init__.py │ └── boring_model.py └── test_lightning.py └── unittests ├── __init__.py ├── _helpers ├── __init__.py ├── testers.py └── wrappers.py ├── audio ├── __init__.py ├── test_c_si_snr.py ├── test_dnsmos.py ├── test_nisqa.py ├── test_pesq.py ├── test_pit.py ├── test_sa_sdr.py ├── test_sdr.py ├── test_si_sdr.py ├── test_si_snr.py ├── test_snr.py ├── test_srmr.py └── test_stoi.py ├── bases ├── __init__.py ├── test_aggregation.py ├── test_collections.py ├── test_composition.py ├── test_ddp.py ├── test_hashing.py ├── test_metric.py └── test_saving_loading.py ├── classification ├── __init__.py ├── _inputs.py ├── test_accuracy.py ├── test_auc.py ├── test_auroc.py ├── test_average_precision.py ├── test_calibration_error.py ├── test_cohen_kappa.py ├── test_confusion_matrix.py ├── test_eer.py ├── test_exact_match.py ├── test_f_beta.py ├── test_group_fairness.py ├── test_hamming_distance.py ├── test_hinge.py ├── test_jaccard.py ├── test_logauc.py ├── test_matthews_corrcoef.py ├── test_negative_predictive_value.py ├── test_precision_fixed_recall.py ├── test_precision_recall.py ├── test_precision_recall_curve.py ├── test_ranking.py ├── test_recall_fixed_precision.py ├── test_roc.py ├── test_sensitivity_specificity.py ├── test_specificity.py ├── test_specificity_sensitivity.py └── test_stat_scores.py ├── clustering ├── __init__.py ├── _inputs.py ├── test_adjusted_mutual_info_score.py ├── test_adjusted_rand_score.py ├── test_calinski_harabasz_score.py ├── test_cluster_accuracy.py ├── test_davies_bouldin_score.py ├── test_dunn_index.py ├── test_fowlkes_mallows_index.py ├── test_homogeneity_completeness_v_measure.py ├── test_mutual_info_score.py ├── test_normalized_mutual_info_score.py ├── test_rand_score.py └── test_utils.py ├── conftest.py ├── deprecations ├── __init__.py └── root_class_imports.py ├── detection ├── __init__.py ├── test_intersection.py ├── test_map.py ├── test_modified_panoptic_quality.py └── test_panoptic_quality.py ├── image ├── __init__.py ├── test_arniqa.py ├── test_csi.py ├── test_d_lambda.py ├── test_d_s.py ├── test_dists.py ├── test_ergas.py ├── test_fid.py ├── test_image_gradients.py ├── test_inception.py ├── test_kid.py ├── test_lpips.py ├── test_mifid.py ├── test_ms_ssim.py ├── test_perceptual_path_length.py ├── test_psnr.py ├── test_psnrb.py ├── test_qnr.py ├── test_rase.py ├── test_rmse_sw.py ├── test_sam.py ├── test_scc.py ├── test_ssim.py ├── test_tv.py ├── test_uqi.py └── test_vif.py ├── multimodal ├── __init__.py ├── test_clip_iqa.py ├── test_clip_score.py └── test_lve.py ├── nominal ├── __init__.py ├── test_cramers.py ├── test_fleiss_kappa.py ├── test_pearson.py ├── test_theils_u.py └── test_tschuprows.py ├── pairwise ├── __init__.py └── test_pairwise_distance.py ├── regression ├── __init__.py ├── test_concordance.py ├── test_cosine_similarity.py ├── test_crps.py ├── test_explained_variance.py ├── test_js_divergence.py ├── test_kendall.py ├── test_kl_divergence.py ├── test_log_cosh_error.py ├── test_mean_error.py ├── test_minkowski_distance.py ├── test_pearson.py ├── test_r2.py ├── test_rse.py ├── test_spearman.py └── test_tweedie_deviance.py ├── retrieval ├── __init__.py ├── _inputs.py ├── helpers.py ├── test_auroc.py ├── test_fallout.py ├── test_hit_rate.py ├── test_map.py ├── test_mrr.py ├── test_ndcg.py ├── test_precision.py ├── test_precision_recall_curve.py ├── test_r_precision.py └── test_recall.py ├── segmentation ├── __init__.py ├── inputs.py ├── test_dice.py ├── test_generalized_dice_score.py ├── test_hausdorff_distance.py ├── test_mean_iou.py └── test_utils.py ├── shape ├── __init__.py └── test_procrustes.py ├── text ├── __init__.py ├── _helpers.py ├── _inputs.py ├── test_bertscore.py ├── test_bleu.py ├── test_cer.py ├── test_chrf.py ├── test_edit.py ├── test_eed.py ├── test_infolm.py ├── test_mer.py ├── test_perplexity.py ├── test_rouge.py ├── test_sacre_bleu.py ├── test_squad.py ├── test_ter.py ├── test_wer.py ├── test_wil.py └── test_wip.py ├── utilities ├── test_plot.py └── test_utilities.py ├── video ├── __init__.py └── test_vmaf.py └── wrappers ├── __init__.py ├── test_bootstrapping.py ├── test_classwise.py ├── test_feature_share.py ├── test_minmax.py ├── test_multioutput.py ├── test_multitask.py ├── test_running.py ├── test_tracker.py └── test_transformations.py /.azure/gpu-integrations.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.azure/gpu-integrations.yml -------------------------------------------------------------------------------- /.azure/gpu-nuke-cache.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.azure/gpu-nuke-cache.yml -------------------------------------------------------------------------------- /.azure/gpu-unittests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.azure/gpu-unittests.yml -------------------------------------------------------------------------------- /.codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.codecov.yml -------------------------------------------------------------------------------- /.devcontainer/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.devcontainer/Dockerfile -------------------------------------------------------------------------------- /.devcontainer/devcontainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.devcontainer/devcontainer.json -------------------------------------------------------------------------------- /.github/CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/CODEOWNERS -------------------------------------------------------------------------------- /.github/CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/CONTRIBUTING.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/documentation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/ISSUE_TEMPLATE/documentation.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/actions/pull-caches/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/actions/pull-caches/action.yml -------------------------------------------------------------------------------- /.github/actions/push-caches/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/actions/push-caches/action.yml -------------------------------------------------------------------------------- /.github/assistant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/assistant.py -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/labeling-config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/labeling-config.yml -------------------------------------------------------------------------------- /.github/markdown-links-config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/markdown-links-config.json -------------------------------------------------------------------------------- /.github/mergify.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/mergify.yml -------------------------------------------------------------------------------- /.github/stale.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/stale.yml -------------------------------------------------------------------------------- /.github/workflows/_focus-diff.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/workflows/_focus-diff.yml -------------------------------------------------------------------------------- /.github/workflows/ci-checks.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/workflows/ci-checks.yml -------------------------------------------------------------------------------- /.github/workflows/ci-integrate.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/workflows/ci-integrate.yml -------------------------------------------------------------------------------- /.github/workflows/ci-rtfd.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/workflows/ci-rtfd.yml -------------------------------------------------------------------------------- /.github/workflows/ci-tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/workflows/ci-tests.yml -------------------------------------------------------------------------------- /.github/workflows/clear-cache.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/workflows/clear-cache.yml -------------------------------------------------------------------------------- /.github/workflows/cmd-help.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/workflows/cmd-help.yml -------------------------------------------------------------------------------- /.github/workflows/cmd-rebase.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/workflows/cmd-rebase.yml -------------------------------------------------------------------------------- /.github/workflows/docker-build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/workflows/docker-build.yml -------------------------------------------------------------------------------- /.github/workflows/docs-build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/workflows/docs-build.yml -------------------------------------------------------------------------------- /.github/workflows/greetings.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/workflows/greetings.yml -------------------------------------------------------------------------------- /.github/workflows/labeler.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/workflows/labeler.yml -------------------------------------------------------------------------------- /.github/workflows/publish-pkg.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/workflows/publish-pkg.yml -------------------------------------------------------------------------------- /.github/workflows/slash-cmd-dispatch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.github/workflows/slash-cmd-dispatch.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.gitignore -------------------------------------------------------------------------------- /.lightning/workflows/integrations.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.lightning/workflows/integrations.yml -------------------------------------------------------------------------------- /.lightning/workflows/uinttests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.lightning/workflows/uinttests.yml -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/README.md -------------------------------------------------------------------------------- /_samples/bert_score-own_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/_samples/bert_score-own_model.py -------------------------------------------------------------------------------- /_samples/detection_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/_samples/detection_map.py -------------------------------------------------------------------------------- /_samples/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/_samples/plotting.py -------------------------------------------------------------------------------- /_samples/rouge_score-own_normalizer_and_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/_samples/rouge_score-own_normalizer_and_tokenizer.py -------------------------------------------------------------------------------- /dockers/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/dockers/README.md -------------------------------------------------------------------------------- /dockers/ubuntu-cuda/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/dockers/ubuntu-cuda/Dockerfile -------------------------------------------------------------------------------- /docs/.build_docs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/.build_docs.sh -------------------------------------------------------------------------------- /docs/.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/.readthedocs.yaml -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/paper_JOSS/paper.bib: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/paper_JOSS/paper.bib -------------------------------------------------------------------------------- /docs/paper_JOSS/paper.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/paper_JOSS/paper.md -------------------------------------------------------------------------------- /docs/redirect.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/redirect.html -------------------------------------------------------------------------------- /docs/rtfd-build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/rtfd-build.sh -------------------------------------------------------------------------------- /docs/source/_static/copybutton.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/_static/copybutton.js -------------------------------------------------------------------------------- /docs/source/_static/css/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/_static/css/custom.css -------------------------------------------------------------------------------- /docs/source/_static/images/icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/_static/images/icon.png -------------------------------------------------------------------------------- /docs/source/_static/images/icon.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/_static/images/icon.svg -------------------------------------------------------------------------------- /docs/source/_static/images/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/_static/images/logo.png -------------------------------------------------------------------------------- /docs/source/_static/images/logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/_static/images/logo.svg -------------------------------------------------------------------------------- /docs/source/_static/images/logo_light.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/_static/images/logo_light.svg -------------------------------------------------------------------------------- /docs/source/_static/images/plot_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/_static/images/plot_example.png -------------------------------------------------------------------------------- /docs/source/_templates/autosummary/module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/_templates/autosummary/module.rst -------------------------------------------------------------------------------- /docs/source/_templates/classtemplate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/_templates/classtemplate.rst -------------------------------------------------------------------------------- /docs/source/_templates/layout.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/_templates/layout.html -------------------------------------------------------------------------------- /docs/source/_templates/theme_variables.jinja: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/_templates/theme_variables.jinja -------------------------------------------------------------------------------- /docs/source/aggregation/cat.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/aggregation/cat.rst -------------------------------------------------------------------------------- /docs/source/aggregation/max.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/aggregation/max.rst -------------------------------------------------------------------------------- /docs/source/aggregation/mean.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/aggregation/mean.rst -------------------------------------------------------------------------------- /docs/source/aggregation/min.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/aggregation/min.rst -------------------------------------------------------------------------------- /docs/source/aggregation/running_mean.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/aggregation/running_mean.rst -------------------------------------------------------------------------------- /docs/source/aggregation/running_sum.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/aggregation/running_sum.rst -------------------------------------------------------------------------------- /docs/source/aggregation/sum.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/aggregation/sum.rst -------------------------------------------------------------------------------- /docs/source/all-metrics.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/all-metrics.rst -------------------------------------------------------------------------------- /docs/source/audio/non_intrusive_speech_quality_assessment.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/audio/non_intrusive_speech_quality_assessment.rst -------------------------------------------------------------------------------- /docs/source/audio/perceptual_evaluation_speech_quality.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/audio/perceptual_evaluation_speech_quality.rst -------------------------------------------------------------------------------- /docs/source/audio/permutation_invariant_training.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/audio/permutation_invariant_training.rst -------------------------------------------------------------------------------- /docs/source/audio/scale_invariant_signal_distortion_ratio.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/audio/scale_invariant_signal_distortion_ratio.rst -------------------------------------------------------------------------------- /docs/source/audio/scale_invariant_signal_noise_ratio.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/audio/scale_invariant_signal_noise_ratio.rst -------------------------------------------------------------------------------- /docs/source/audio/short_time_objective_intelligibility.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/audio/short_time_objective_intelligibility.rst -------------------------------------------------------------------------------- /docs/source/audio/signal_distortion_ratio.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/audio/signal_distortion_ratio.rst -------------------------------------------------------------------------------- /docs/source/audio/signal_noise_ratio.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/audio/signal_noise_ratio.rst -------------------------------------------------------------------------------- /docs/source/classification/accuracy.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/accuracy.rst -------------------------------------------------------------------------------- /docs/source/classification/auroc.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/auroc.rst -------------------------------------------------------------------------------- /docs/source/classification/average_precision.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/average_precision.rst -------------------------------------------------------------------------------- /docs/source/classification/calibration_error.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/calibration_error.rst -------------------------------------------------------------------------------- /docs/source/classification/cohen_kappa.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/cohen_kappa.rst -------------------------------------------------------------------------------- /docs/source/classification/confusion_matrix.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/confusion_matrix.rst -------------------------------------------------------------------------------- /docs/source/classification/coverage_error.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/coverage_error.rst -------------------------------------------------------------------------------- /docs/source/classification/eer.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/eer.rst -------------------------------------------------------------------------------- /docs/source/classification/exact_match.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/exact_match.rst -------------------------------------------------------------------------------- /docs/source/classification/f1_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/f1_score.rst -------------------------------------------------------------------------------- /docs/source/classification/fbeta_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/fbeta_score.rst -------------------------------------------------------------------------------- /docs/source/classification/group_fairness.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/group_fairness.rst -------------------------------------------------------------------------------- /docs/source/classification/hamming_distance.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/hamming_distance.rst -------------------------------------------------------------------------------- /docs/source/classification/hinge_loss.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/hinge_loss.rst -------------------------------------------------------------------------------- /docs/source/classification/jaccard_index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/jaccard_index.rst -------------------------------------------------------------------------------- /docs/source/classification/label_ranking_loss.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/label_ranking_loss.rst -------------------------------------------------------------------------------- /docs/source/classification/logauc.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/logauc.rst -------------------------------------------------------------------------------- /docs/source/classification/matthews_corr_coef.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/matthews_corr_coef.rst -------------------------------------------------------------------------------- /docs/source/classification/negative_predictive_value.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/negative_predictive_value.rst -------------------------------------------------------------------------------- /docs/source/classification/precision.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/precision.rst -------------------------------------------------------------------------------- /docs/source/classification/precision_at_fixed_recall.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/precision_at_fixed_recall.rst -------------------------------------------------------------------------------- /docs/source/classification/precision_recall_curve.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/precision_recall_curve.rst -------------------------------------------------------------------------------- /docs/source/classification/recall.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/recall.rst -------------------------------------------------------------------------------- /docs/source/classification/recall_at_fixed_precision.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/recall_at_fixed_precision.rst -------------------------------------------------------------------------------- /docs/source/classification/roc.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/roc.rst -------------------------------------------------------------------------------- /docs/source/classification/sensitivity_at_specificity.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/sensitivity_at_specificity.rst -------------------------------------------------------------------------------- /docs/source/classification/specificity.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/specificity.rst -------------------------------------------------------------------------------- /docs/source/classification/specificity_at_sensitivity.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/specificity_at_sensitivity.rst -------------------------------------------------------------------------------- /docs/source/classification/stat_scores.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/classification/stat_scores.rst -------------------------------------------------------------------------------- /docs/source/clustering/adjusted_mutual_info_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/clustering/adjusted_mutual_info_score.rst -------------------------------------------------------------------------------- /docs/source/clustering/adjusted_rand_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/clustering/adjusted_rand_score.rst -------------------------------------------------------------------------------- /docs/source/clustering/calinski_harabasz_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/clustering/calinski_harabasz_score.rst -------------------------------------------------------------------------------- /docs/source/clustering/cluster_accuracy.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/clustering/cluster_accuracy.rst -------------------------------------------------------------------------------- /docs/source/clustering/completeness_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/clustering/completeness_score.rst -------------------------------------------------------------------------------- /docs/source/clustering/davies_bouldin_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/clustering/davies_bouldin_score.rst -------------------------------------------------------------------------------- /docs/source/clustering/dunn_index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/clustering/dunn_index.rst -------------------------------------------------------------------------------- /docs/source/clustering/fowlkes_mallows_index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/clustering/fowlkes_mallows_index.rst -------------------------------------------------------------------------------- /docs/source/clustering/homogeneity_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/clustering/homogeneity_score.rst -------------------------------------------------------------------------------- /docs/source/clustering/mutual_info_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/clustering/mutual_info_score.rst -------------------------------------------------------------------------------- /docs/source/clustering/normalized_mutual_info_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/clustering/normalized_mutual_info_score.rst -------------------------------------------------------------------------------- /docs/source/clustering/rand_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/clustering/rand_score.rst -------------------------------------------------------------------------------- /docs/source/clustering/v_measure_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/clustering/v_measure_score.rst -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/detection/complete_intersection_over_union.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/detection/complete_intersection_over_union.rst -------------------------------------------------------------------------------- /docs/source/detection/distance_intersection_over_union.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/detection/distance_intersection_over_union.rst -------------------------------------------------------------------------------- /docs/source/detection/generalized_intersection_over_union.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/detection/generalized_intersection_over_union.rst -------------------------------------------------------------------------------- /docs/source/detection/intersection_over_union.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/detection/intersection_over_union.rst -------------------------------------------------------------------------------- /docs/source/detection/mean_average_precision.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/detection/mean_average_precision.rst -------------------------------------------------------------------------------- /docs/source/detection/modified_panoptic_quality.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/detection/modified_panoptic_quality.rst -------------------------------------------------------------------------------- /docs/source/detection/panoptic_quality.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/detection/panoptic_quality.rst -------------------------------------------------------------------------------- /docs/source/governance.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/governance.rst -------------------------------------------------------------------------------- /docs/source/image/arniqa.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/arniqa.rst -------------------------------------------------------------------------------- /docs/source/image/dists.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/dists.rst -------------------------------------------------------------------------------- /docs/source/image/frechet_inception_distance.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/frechet_inception_distance.rst -------------------------------------------------------------------------------- /docs/source/image/image_gradients.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/image_gradients.rst -------------------------------------------------------------------------------- /docs/source/image/inception_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/inception_score.rst -------------------------------------------------------------------------------- /docs/source/image/kernel_inception_distance.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/kernel_inception_distance.rst -------------------------------------------------------------------------------- /docs/source/image/mifid.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/mifid.rst -------------------------------------------------------------------------------- /docs/source/image/multi_scale_structural_similarity.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/multi_scale_structural_similarity.rst -------------------------------------------------------------------------------- /docs/source/image/peak_signal_noise_ratio.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/peak_signal_noise_ratio.rst -------------------------------------------------------------------------------- /docs/source/image/peak_signal_to_noise_with_block.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/peak_signal_to_noise_with_block.rst -------------------------------------------------------------------------------- /docs/source/image/perceptual_path_length.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/perceptual_path_length.rst -------------------------------------------------------------------------------- /docs/source/image/quality_with_no_reference.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/quality_with_no_reference.rst -------------------------------------------------------------------------------- /docs/source/image/relative_average_spectral_error.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/relative_average_spectral_error.rst -------------------------------------------------------------------------------- /docs/source/image/spatial_correlation_coefficient.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/spatial_correlation_coefficient.rst -------------------------------------------------------------------------------- /docs/source/image/spatial_distortion_index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/spatial_distortion_index.rst -------------------------------------------------------------------------------- /docs/source/image/spectral_angle_mapper.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/spectral_angle_mapper.rst -------------------------------------------------------------------------------- /docs/source/image/spectral_distortion_index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/spectral_distortion_index.rst -------------------------------------------------------------------------------- /docs/source/image/structural_similarity.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/structural_similarity.rst -------------------------------------------------------------------------------- /docs/source/image/total_variation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/total_variation.rst -------------------------------------------------------------------------------- /docs/source/image/universal_image_quality_index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/universal_image_quality_index.rst -------------------------------------------------------------------------------- /docs/source/image/visual_information_fidelity.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/image/visual_information_fidelity.rst -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/links.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/links.rst -------------------------------------------------------------------------------- /docs/source/multimodal/clip_iqa.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/multimodal/clip_iqa.rst -------------------------------------------------------------------------------- /docs/source/multimodal/clip_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/multimodal/clip_score.rst -------------------------------------------------------------------------------- /docs/source/multimodal/lve.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/multimodal/lve.rst -------------------------------------------------------------------------------- /docs/source/nominal/cramers_v.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/nominal/cramers_v.rst -------------------------------------------------------------------------------- /docs/source/nominal/fleiss_kappa.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/nominal/fleiss_kappa.rst -------------------------------------------------------------------------------- /docs/source/nominal/pearsons_contingency_coefficient.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/nominal/pearsons_contingency_coefficient.rst -------------------------------------------------------------------------------- /docs/source/nominal/theils_u.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/nominal/theils_u.rst -------------------------------------------------------------------------------- /docs/source/nominal/tschuprows_t.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/nominal/tschuprows_t.rst -------------------------------------------------------------------------------- /docs/source/pages/implement.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pages/implement.rst -------------------------------------------------------------------------------- /docs/source/pages/lightning.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pages/lightning.rst -------------------------------------------------------------------------------- /docs/source/pages/overview.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pages/overview.rst -------------------------------------------------------------------------------- /docs/source/pages/plotting.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pages/plotting.rst -------------------------------------------------------------------------------- /docs/source/pages/quickstart.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pages/quickstart.rst -------------------------------------------------------------------------------- /docs/source/pairwise/cosine_similarity.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pairwise/cosine_similarity.rst -------------------------------------------------------------------------------- /docs/source/pairwise/euclidean_distance.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pairwise/euclidean_distance.rst -------------------------------------------------------------------------------- /docs/source/pairwise/linear_similarity.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pairwise/linear_similarity.rst -------------------------------------------------------------------------------- /docs/source/pairwise/manhattan_distance.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pairwise/manhattan_distance.rst -------------------------------------------------------------------------------- /docs/source/pairwise/minkowski_distance.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pairwise/minkowski_distance.rst -------------------------------------------------------------------------------- /docs/source/pyplots/binary_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pyplots/binary_accuracy.py -------------------------------------------------------------------------------- /docs/source/pyplots/binary_accuracy_multistep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pyplots/binary_accuracy_multistep.py -------------------------------------------------------------------------------- /docs/source/pyplots/collection_binary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pyplots/collection_binary.py -------------------------------------------------------------------------------- /docs/source/pyplots/collection_binary_together.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pyplots/collection_binary_together.py -------------------------------------------------------------------------------- /docs/source/pyplots/confusion_matrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pyplots/confusion_matrix.py -------------------------------------------------------------------------------- /docs/source/pyplots/multiclass_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pyplots/multiclass_accuracy.py -------------------------------------------------------------------------------- /docs/source/pyplots/tracker_binary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/pyplots/tracker_binary.py -------------------------------------------------------------------------------- /docs/source/references/metric.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/references/metric.rst -------------------------------------------------------------------------------- /docs/source/references/utilities.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/references/utilities.rst -------------------------------------------------------------------------------- /docs/source/regression/concordance_corr_coef.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/concordance_corr_coef.rst -------------------------------------------------------------------------------- /docs/source/regression/cosine_similarity.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/cosine_similarity.rst -------------------------------------------------------------------------------- /docs/source/regression/critical_success_index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/critical_success_index.rst -------------------------------------------------------------------------------- /docs/source/regression/crps.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/crps.rst -------------------------------------------------------------------------------- /docs/source/regression/explained_variance.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/explained_variance.rst -------------------------------------------------------------------------------- /docs/source/regression/js_divergence.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/js_divergence.rst -------------------------------------------------------------------------------- /docs/source/regression/kendall_rank_corr_coef.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/kendall_rank_corr_coef.rst -------------------------------------------------------------------------------- /docs/source/regression/kl_divergence.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/kl_divergence.rst -------------------------------------------------------------------------------- /docs/source/regression/log_cosh_error.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/log_cosh_error.rst -------------------------------------------------------------------------------- /docs/source/regression/mean_absolute_error.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/mean_absolute_error.rst -------------------------------------------------------------------------------- /docs/source/regression/mean_absolute_percentage_error.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/mean_absolute_percentage_error.rst -------------------------------------------------------------------------------- /docs/source/regression/mean_squared_error.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/mean_squared_error.rst -------------------------------------------------------------------------------- /docs/source/regression/mean_squared_log_error.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/mean_squared_log_error.rst -------------------------------------------------------------------------------- /docs/source/regression/minkowski_distance.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/minkowski_distance.rst -------------------------------------------------------------------------------- /docs/source/regression/normalized_root_mean_squared_error.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/normalized_root_mean_squared_error.rst -------------------------------------------------------------------------------- /docs/source/regression/pearson_corr_coef.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/pearson_corr_coef.rst -------------------------------------------------------------------------------- /docs/source/regression/r2_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/r2_score.rst -------------------------------------------------------------------------------- /docs/source/regression/rse.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/rse.rst -------------------------------------------------------------------------------- /docs/source/regression/spearman_corr_coef.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/spearman_corr_coef.rst -------------------------------------------------------------------------------- /docs/source/regression/tweedie_deviance_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/regression/tweedie_deviance_score.rst -------------------------------------------------------------------------------- /docs/source/retrieval/auroc.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/retrieval/auroc.rst -------------------------------------------------------------------------------- /docs/source/retrieval/fall_out.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/retrieval/fall_out.rst -------------------------------------------------------------------------------- /docs/source/retrieval/hit_rate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/retrieval/hit_rate.rst -------------------------------------------------------------------------------- /docs/source/retrieval/map.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/retrieval/map.rst -------------------------------------------------------------------------------- /docs/source/retrieval/mrr.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/retrieval/mrr.rst -------------------------------------------------------------------------------- /docs/source/retrieval/normalized_dcg.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/retrieval/normalized_dcg.rst -------------------------------------------------------------------------------- /docs/source/retrieval/precision.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/retrieval/precision.rst -------------------------------------------------------------------------------- /docs/source/retrieval/precision_recall_curve.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/retrieval/precision_recall_curve.rst -------------------------------------------------------------------------------- /docs/source/retrieval/r_precision.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/retrieval/r_precision.rst -------------------------------------------------------------------------------- /docs/source/retrieval/recall.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/retrieval/recall.rst -------------------------------------------------------------------------------- /docs/source/segmentation/dice.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/segmentation/dice.rst -------------------------------------------------------------------------------- /docs/source/segmentation/generalized_dice.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/segmentation/generalized_dice.rst -------------------------------------------------------------------------------- /docs/source/segmentation/hausdorff_distance.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/segmentation/hausdorff_distance.rst -------------------------------------------------------------------------------- /docs/source/segmentation/mean_iou.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/segmentation/mean_iou.rst -------------------------------------------------------------------------------- /docs/source/shape/procrustes.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/shape/procrustes.rst -------------------------------------------------------------------------------- /docs/source/text/bert_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/bert_score.rst -------------------------------------------------------------------------------- /docs/source/text/bleu_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/bleu_score.rst -------------------------------------------------------------------------------- /docs/source/text/char_error_rate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/char_error_rate.rst -------------------------------------------------------------------------------- /docs/source/text/chrf_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/chrf_score.rst -------------------------------------------------------------------------------- /docs/source/text/edit.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/edit.rst -------------------------------------------------------------------------------- /docs/source/text/extended_edit_distance.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/extended_edit_distance.rst -------------------------------------------------------------------------------- /docs/source/text/infolm.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/infolm.rst -------------------------------------------------------------------------------- /docs/source/text/match_error_rate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/match_error_rate.rst -------------------------------------------------------------------------------- /docs/source/text/perplexity.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/perplexity.rst -------------------------------------------------------------------------------- /docs/source/text/rouge_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/rouge_score.rst -------------------------------------------------------------------------------- /docs/source/text/sacre_bleu_score.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/sacre_bleu_score.rst -------------------------------------------------------------------------------- /docs/source/text/squad.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/squad.rst -------------------------------------------------------------------------------- /docs/source/text/translation_edit_rate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/translation_edit_rate.rst -------------------------------------------------------------------------------- /docs/source/text/word_error_rate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/word_error_rate.rst -------------------------------------------------------------------------------- /docs/source/text/word_info_lost.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/word_info_lost.rst -------------------------------------------------------------------------------- /docs/source/text/word_info_preserved.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/text/word_info_preserved.rst -------------------------------------------------------------------------------- /docs/source/video/vmaf.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/video/vmaf.rst -------------------------------------------------------------------------------- /docs/source/wrappers/bootstrapper.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/wrappers/bootstrapper.rst -------------------------------------------------------------------------------- /docs/source/wrappers/classwise_wrapper.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/wrappers/classwise_wrapper.rst -------------------------------------------------------------------------------- /docs/source/wrappers/feature_share.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/wrappers/feature_share.rst -------------------------------------------------------------------------------- /docs/source/wrappers/metric_tracker.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/wrappers/metric_tracker.rst -------------------------------------------------------------------------------- /docs/source/wrappers/min_max.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/wrappers/min_max.rst -------------------------------------------------------------------------------- /docs/source/wrappers/multi_output_wrapper.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/wrappers/multi_output_wrapper.rst -------------------------------------------------------------------------------- /docs/source/wrappers/multi_task_wrapper.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/wrappers/multi_task_wrapper.rst -------------------------------------------------------------------------------- /docs/source/wrappers/running.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/wrappers/running.rst -------------------------------------------------------------------------------- /docs/source/wrappers/transformations.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/docs/source/wrappers/transformations.rst -------------------------------------------------------------------------------- /examples/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/examples/README.rst -------------------------------------------------------------------------------- /examples/audio/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/examples/audio/README.rst -------------------------------------------------------------------------------- /examples/audio/pesq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/examples/audio/pesq.py -------------------------------------------------------------------------------- /examples/audio/signal_to_noise_ratio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/examples/audio/signal_to_noise_ratio.py -------------------------------------------------------------------------------- /examples/audio/text_to_speech.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/examples/audio/text_to_speech.py -------------------------------------------------------------------------------- /examples/image/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/examples/image/README.rst -------------------------------------------------------------------------------- /examples/image/clip_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/examples/image/clip_score.py -------------------------------------------------------------------------------- /examples/image/spatial_correlation_coef.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/examples/image/spatial_correlation_coef.py -------------------------------------------------------------------------------- /examples/text/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/examples/text/README.rst -------------------------------------------------------------------------------- /examples/text/bertscore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/examples/text/bertscore.py -------------------------------------------------------------------------------- /examples/text/perplexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/examples/text/perplexity.py -------------------------------------------------------------------------------- /examples/text/rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/examples/text/rouge.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | -r requirements/base.txt 2 | -------------------------------------------------------------------------------- /requirements/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/README.md -------------------------------------------------------------------------------- /requirements/_ci.txt: -------------------------------------------------------------------------------- 1 | py-tree 2 | packaging 3 | fire 4 | wget 5 | -------------------------------------------------------------------------------- /requirements/_devel.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/_devel.txt -------------------------------------------------------------------------------- /requirements/_docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/_docs.txt -------------------------------------------------------------------------------- /requirements/_doctest.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/_doctest.txt -------------------------------------------------------------------------------- /requirements/_integrate.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/_integrate.txt -------------------------------------------------------------------------------- /requirements/_tests.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/_tests.txt -------------------------------------------------------------------------------- /requirements/audio.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/audio.txt -------------------------------------------------------------------------------- /requirements/audio_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/audio_test.txt -------------------------------------------------------------------------------- /requirements/base.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/base.txt -------------------------------------------------------------------------------- /requirements/classification_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/classification_test.txt -------------------------------------------------------------------------------- /requirements/clustering.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/clustering.txt -------------------------------------------------------------------------------- /requirements/clustering_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/clustering_test.txt -------------------------------------------------------------------------------- /requirements/debug.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/debug.txt -------------------------------------------------------------------------------- /requirements/detection.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/detection.txt -------------------------------------------------------------------------------- /requirements/detection_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/detection_test.txt -------------------------------------------------------------------------------- /requirements/image.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/image.txt -------------------------------------------------------------------------------- /requirements/image_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/image_test.txt -------------------------------------------------------------------------------- /requirements/integrate.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements/multimodal.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/multimodal.txt -------------------------------------------------------------------------------- /requirements/nominal_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/nominal_test.txt -------------------------------------------------------------------------------- /requirements/regression_test.txt: -------------------------------------------------------------------------------- 1 | permetrics==2.0.0 2 | properscoring ==0.1 3 | -------------------------------------------------------------------------------- /requirements/segmentation_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/segmentation_test.txt -------------------------------------------------------------------------------- /requirements/text.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/text.txt -------------------------------------------------------------------------------- /requirements/text_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/text_test.txt -------------------------------------------------------------------------------- /requirements/typing.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/typing.txt -------------------------------------------------------------------------------- /requirements/video.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/video.txt -------------------------------------------------------------------------------- /requirements/visual.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/requirements/visual.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/setup.py -------------------------------------------------------------------------------- /src/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/conftest.py -------------------------------------------------------------------------------- /src/torchmetrics/__about__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/__about__.py -------------------------------------------------------------------------------- /src/torchmetrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/aggregation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/aggregation.py -------------------------------------------------------------------------------- /src/torchmetrics/audio/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/audio/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/audio/_deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/audio/_deprecated.py -------------------------------------------------------------------------------- /src/torchmetrics/audio/dnsmos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/audio/dnsmos.py -------------------------------------------------------------------------------- /src/torchmetrics/audio/nisqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/audio/nisqa.py -------------------------------------------------------------------------------- /src/torchmetrics/audio/pesq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/audio/pesq.py -------------------------------------------------------------------------------- /src/torchmetrics/audio/pit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/audio/pit.py -------------------------------------------------------------------------------- /src/torchmetrics/audio/sdr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/audio/sdr.py -------------------------------------------------------------------------------- /src/torchmetrics/audio/snr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/audio/snr.py -------------------------------------------------------------------------------- /src/torchmetrics/audio/srmr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/audio/srmr.py -------------------------------------------------------------------------------- /src/torchmetrics/audio/stoi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/audio/stoi.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/accuracy.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/auroc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/auroc.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/average_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/average_precision.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/base.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/calibration_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/calibration_error.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/cohen_kappa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/cohen_kappa.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/confusion_matrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/confusion_matrix.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/eer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/eer.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/exact_match.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/exact_match.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/f_beta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/f_beta.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/group_fairness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/group_fairness.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/hamming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/hamming.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/hinge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/hinge.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/jaccard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/jaccard.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/logauc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/logauc.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/matthews_corrcoef.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/matthews_corrcoef.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/negative_predictive_value.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/negative_predictive_value.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/precision_fixed_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/precision_fixed_recall.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/precision_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/precision_recall.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/precision_recall_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/precision_recall_curve.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/ranking.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/recall_fixed_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/recall_fixed_precision.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/roc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/roc.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/sensitivity_specificity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/sensitivity_specificity.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/specificity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/specificity.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/specificity_sensitivity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/specificity_sensitivity.py -------------------------------------------------------------------------------- /src/torchmetrics/classification/stat_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/classification/stat_scores.py -------------------------------------------------------------------------------- /src/torchmetrics/clustering/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/clustering/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/clustering/adjusted_mutual_info_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/clustering/adjusted_mutual_info_score.py -------------------------------------------------------------------------------- /src/torchmetrics/clustering/adjusted_rand_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/clustering/adjusted_rand_score.py -------------------------------------------------------------------------------- /src/torchmetrics/clustering/calinski_harabasz_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/clustering/calinski_harabasz_score.py -------------------------------------------------------------------------------- /src/torchmetrics/clustering/cluster_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/clustering/cluster_accuracy.py -------------------------------------------------------------------------------- /src/torchmetrics/clustering/davies_bouldin_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/clustering/davies_bouldin_score.py -------------------------------------------------------------------------------- /src/torchmetrics/clustering/dunn_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/clustering/dunn_index.py -------------------------------------------------------------------------------- /src/torchmetrics/clustering/fowlkes_mallows_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/clustering/fowlkes_mallows_index.py -------------------------------------------------------------------------------- /src/torchmetrics/clustering/mutual_info_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/clustering/mutual_info_score.py -------------------------------------------------------------------------------- /src/torchmetrics/clustering/normalized_mutual_info_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/clustering/normalized_mutual_info_score.py -------------------------------------------------------------------------------- /src/torchmetrics/clustering/rand_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/clustering/rand_score.py -------------------------------------------------------------------------------- /src/torchmetrics/collections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/collections.py -------------------------------------------------------------------------------- /src/torchmetrics/detection/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/detection/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/detection/_deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/detection/_deprecated.py -------------------------------------------------------------------------------- /src/torchmetrics/detection/_mean_ap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/detection/_mean_ap.py -------------------------------------------------------------------------------- /src/torchmetrics/detection/ciou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/detection/ciou.py -------------------------------------------------------------------------------- /src/torchmetrics/detection/diou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/detection/diou.py -------------------------------------------------------------------------------- /src/torchmetrics/detection/giou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/detection/giou.py -------------------------------------------------------------------------------- /src/torchmetrics/detection/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/detection/helpers.py -------------------------------------------------------------------------------- /src/torchmetrics/detection/iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/detection/iou.py -------------------------------------------------------------------------------- /src/torchmetrics/detection/mean_ap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/detection/mean_ap.py -------------------------------------------------------------------------------- /src/torchmetrics/detection/panoptic_qualities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/detection/panoptic_qualities.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/audio/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/audio/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/audio/_deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/audio/_deprecated.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/audio/dnsmos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/audio/dnsmos.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/audio/nisqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/audio/nisqa.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/audio/pesq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/audio/pesq.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/audio/pit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/audio/pit.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/audio/sdr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/audio/sdr.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/audio/snr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/audio/snr.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/audio/srmr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/audio/srmr.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/audio/stoi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/audio/stoi.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/accuracy.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/auroc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/auroc.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/cohen_kappa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/cohen_kappa.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/eer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/eer.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/exact_match.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/exact_match.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/f_beta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/f_beta.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/group_fairness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/group_fairness.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/hamming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/hamming.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/hinge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/hinge.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/jaccard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/jaccard.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/logauc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/logauc.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/ranking.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/roc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/roc.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/specificity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/specificity.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/classification/stat_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/classification/stat_scores.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/clustering/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/clustering/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/clustering/adjusted_rand_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/clustering/adjusted_rand_score.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/clustering/cluster_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/clustering/cluster_accuracy.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/clustering/dunn_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/clustering/dunn_index.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/clustering/mutual_info_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/clustering/mutual_info_score.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/clustering/rand_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/clustering/rand_score.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/clustering/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/clustering/utils.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/detection/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/detection/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/detection/_deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/detection/_deprecated.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/detection/ciou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/detection/ciou.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/detection/diou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/detection/diou.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/detection/giou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/detection/giou.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/detection/iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/detection/iou.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/detection/map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/detection/map.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/_deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/_deprecated.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/arniqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/arniqa.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/d_lambda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/d_lambda.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/d_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/d_s.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/dists.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/dists.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/dists_models/weights.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/dists_models/weights.pt -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/ergas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/ergas.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/gradients.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/gradients.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/lpips.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/lpips_models/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/lpips_models/alex.pth -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/lpips_models/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/lpips_models/squeeze.pth -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/lpips_models/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/lpips_models/vgg.pth -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/psnr.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/psnrb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/psnrb.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/qnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/qnr.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/rase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/rase.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/rmse_sw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/rmse_sw.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/sam.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/scc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/scc.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/ssim.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/tv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/tv.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/uqi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/uqi.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/utils.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/image/vif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/image/vif.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/multimodal/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/multimodal/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/multimodal/clip_iqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/multimodal/clip_iqa.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/multimodal/clip_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/multimodal/clip_score.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/multimodal/lve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/multimodal/lve.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/nominal/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/nominal/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/nominal/cramers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/nominal/cramers.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/nominal/fleiss_kappa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/nominal/fleiss_kappa.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/nominal/pearson.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/nominal/pearson.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/nominal/theils_u.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/nominal/theils_u.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/nominal/tschuprows.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/nominal/tschuprows.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/nominal/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/nominal/utils.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/pairwise/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/pairwise/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/pairwise/cosine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/pairwise/cosine.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/pairwise/euclidean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/pairwise/euclidean.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/pairwise/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/pairwise/helpers.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/pairwise/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/pairwise/linear.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/pairwise/manhattan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/pairwise/manhattan.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/pairwise/minkowski.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/pairwise/minkowski.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/concordance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/concordance.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/crps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/crps.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/csi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/csi.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/js_divergence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/js_divergence.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/kendall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/kendall.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/kl_divergence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/kl_divergence.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/log_cosh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/log_cosh.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/log_mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/log_mse.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/mae.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/mape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/mape.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/minkowski.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/minkowski.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/mse.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/nrmse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/nrmse.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/pearson.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/pearson.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/r2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/r2.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/rse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/rse.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/spearman.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/spearman.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/symmetric_mape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/symmetric_mape.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/tweedie_deviance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/tweedie_deviance.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/utils.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/regression/wmape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/regression/wmape.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/retrieval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/retrieval/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/retrieval/_deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/retrieval/_deprecated.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/retrieval/auroc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/retrieval/auroc.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/retrieval/average_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/retrieval/average_precision.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/retrieval/fall_out.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/retrieval/fall_out.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/retrieval/hit_rate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/retrieval/hit_rate.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/retrieval/ndcg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/retrieval/ndcg.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/retrieval/precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/retrieval/precision.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/retrieval/r_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/retrieval/r_precision.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/retrieval/recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/retrieval/recall.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/retrieval/reciprocal_rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/retrieval/reciprocal_rank.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/segmentation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/segmentation/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/segmentation/dice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/segmentation/dice.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/segmentation/mean_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/segmentation/mean_iou.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/segmentation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/segmentation/utils.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/shape/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/shape/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/shape/procrustes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/shape/procrustes.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/_deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/_deprecated.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/bert.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/bleu.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/cer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/cer.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/chrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/chrf.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/edit.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/eed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/eed.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/helper.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/infolm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/infolm.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/mer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/mer.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/perplexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/perplexity.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/rouge.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/sacre_bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/sacre_bleu.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/squad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/squad.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/ter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/ter.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/wer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/wer.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/wil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/wil.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/text/wip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/text/wip.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/video/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/video/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/functional/video/vmaf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/functional/video/vmaf.py -------------------------------------------------------------------------------- /src/torchmetrics/image/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/image/_deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/_deprecated.py -------------------------------------------------------------------------------- /src/torchmetrics/image/arniqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/arniqa.py -------------------------------------------------------------------------------- /src/torchmetrics/image/d_lambda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/d_lambda.py -------------------------------------------------------------------------------- /src/torchmetrics/image/d_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/d_s.py -------------------------------------------------------------------------------- /src/torchmetrics/image/dists.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/dists.py -------------------------------------------------------------------------------- /src/torchmetrics/image/ergas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/ergas.py -------------------------------------------------------------------------------- /src/torchmetrics/image/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/fid.py -------------------------------------------------------------------------------- /src/torchmetrics/image/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/inception.py -------------------------------------------------------------------------------- /src/torchmetrics/image/kid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/kid.py -------------------------------------------------------------------------------- /src/torchmetrics/image/lpip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/lpip.py -------------------------------------------------------------------------------- /src/torchmetrics/image/mifid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/mifid.py -------------------------------------------------------------------------------- /src/torchmetrics/image/perceptual_path_length.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/perceptual_path_length.py -------------------------------------------------------------------------------- /src/torchmetrics/image/psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/psnr.py -------------------------------------------------------------------------------- /src/torchmetrics/image/psnrb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/psnrb.py -------------------------------------------------------------------------------- /src/torchmetrics/image/qnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/qnr.py -------------------------------------------------------------------------------- /src/torchmetrics/image/rase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/rase.py -------------------------------------------------------------------------------- /src/torchmetrics/image/rmse_sw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/rmse_sw.py -------------------------------------------------------------------------------- /src/torchmetrics/image/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/sam.py -------------------------------------------------------------------------------- /src/torchmetrics/image/scc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/scc.py -------------------------------------------------------------------------------- /src/torchmetrics/image/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/ssim.py -------------------------------------------------------------------------------- /src/torchmetrics/image/tv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/tv.py -------------------------------------------------------------------------------- /src/torchmetrics/image/uqi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/uqi.py -------------------------------------------------------------------------------- /src/torchmetrics/image/vif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/image/vif.py -------------------------------------------------------------------------------- /src/torchmetrics/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/metric.py -------------------------------------------------------------------------------- /src/torchmetrics/multimodal/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/multimodal/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/multimodal/clip_iqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/multimodal/clip_iqa.py -------------------------------------------------------------------------------- /src/torchmetrics/multimodal/clip_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/multimodal/clip_score.py -------------------------------------------------------------------------------- /src/torchmetrics/multimodal/lve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/multimodal/lve.py -------------------------------------------------------------------------------- /src/torchmetrics/nominal/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/nominal/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/nominal/cramers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/nominal/cramers.py -------------------------------------------------------------------------------- /src/torchmetrics/nominal/fleiss_kappa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/nominal/fleiss_kappa.py -------------------------------------------------------------------------------- /src/torchmetrics/nominal/pearson.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/nominal/pearson.py -------------------------------------------------------------------------------- /src/torchmetrics/nominal/theils_u.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/nominal/theils_u.py -------------------------------------------------------------------------------- /src/torchmetrics/nominal/tschuprows.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/nominal/tschuprows.py -------------------------------------------------------------------------------- /src/torchmetrics/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/torchmetrics/regression/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/concordance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/concordance.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/cosine_similarity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/cosine_similarity.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/crps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/crps.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/csi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/csi.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/explained_variance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/explained_variance.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/js_divergence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/js_divergence.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/kendall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/kendall.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/kl_divergence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/kl_divergence.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/log_cosh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/log_cosh.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/log_mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/log_mse.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/mae.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/mape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/mape.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/minkowski.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/minkowski.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/mse.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/nrmse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/nrmse.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/pearson.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/pearson.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/r2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/r2.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/rse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/rse.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/spearman.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/spearman.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/symmetric_mape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/symmetric_mape.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/tweedie_deviance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/tweedie_deviance.py -------------------------------------------------------------------------------- /src/torchmetrics/regression/wmape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/regression/wmape.py -------------------------------------------------------------------------------- /src/torchmetrics/retrieval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/retrieval/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/retrieval/_deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/retrieval/_deprecated.py -------------------------------------------------------------------------------- /src/torchmetrics/retrieval/auroc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/retrieval/auroc.py -------------------------------------------------------------------------------- /src/torchmetrics/retrieval/average_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/retrieval/average_precision.py -------------------------------------------------------------------------------- /src/torchmetrics/retrieval/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/retrieval/base.py -------------------------------------------------------------------------------- /src/torchmetrics/retrieval/fall_out.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/retrieval/fall_out.py -------------------------------------------------------------------------------- /src/torchmetrics/retrieval/hit_rate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/retrieval/hit_rate.py -------------------------------------------------------------------------------- /src/torchmetrics/retrieval/ndcg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/retrieval/ndcg.py -------------------------------------------------------------------------------- /src/torchmetrics/retrieval/precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/retrieval/precision.py -------------------------------------------------------------------------------- /src/torchmetrics/retrieval/precision_recall_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/retrieval/precision_recall_curve.py -------------------------------------------------------------------------------- /src/torchmetrics/retrieval/r_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/retrieval/r_precision.py -------------------------------------------------------------------------------- /src/torchmetrics/retrieval/recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/retrieval/recall.py -------------------------------------------------------------------------------- /src/torchmetrics/retrieval/reciprocal_rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/retrieval/reciprocal_rank.py -------------------------------------------------------------------------------- /src/torchmetrics/segmentation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/segmentation/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/segmentation/dice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/segmentation/dice.py -------------------------------------------------------------------------------- /src/torchmetrics/segmentation/generalized_dice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/segmentation/generalized_dice.py -------------------------------------------------------------------------------- /src/torchmetrics/segmentation/hausdorff_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/segmentation/hausdorff_distance.py -------------------------------------------------------------------------------- /src/torchmetrics/segmentation/mean_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/segmentation/mean_iou.py -------------------------------------------------------------------------------- /src/torchmetrics/shape/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/shape/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/shape/procrustes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/shape/procrustes.py -------------------------------------------------------------------------------- /src/torchmetrics/text/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/text/_deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/_deprecated.py -------------------------------------------------------------------------------- /src/torchmetrics/text/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/bert.py -------------------------------------------------------------------------------- /src/torchmetrics/text/bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/bleu.py -------------------------------------------------------------------------------- /src/torchmetrics/text/cer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/cer.py -------------------------------------------------------------------------------- /src/torchmetrics/text/chrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/chrf.py -------------------------------------------------------------------------------- /src/torchmetrics/text/edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/edit.py -------------------------------------------------------------------------------- /src/torchmetrics/text/eed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/eed.py -------------------------------------------------------------------------------- /src/torchmetrics/text/infolm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/infolm.py -------------------------------------------------------------------------------- /src/torchmetrics/text/mer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/mer.py -------------------------------------------------------------------------------- /src/torchmetrics/text/perplexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/perplexity.py -------------------------------------------------------------------------------- /src/torchmetrics/text/rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/rouge.py -------------------------------------------------------------------------------- /src/torchmetrics/text/sacre_bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/sacre_bleu.py -------------------------------------------------------------------------------- /src/torchmetrics/text/squad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/squad.py -------------------------------------------------------------------------------- /src/torchmetrics/text/ter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/ter.py -------------------------------------------------------------------------------- /src/torchmetrics/text/wer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/wer.py -------------------------------------------------------------------------------- /src/torchmetrics/text/wil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/wil.py -------------------------------------------------------------------------------- /src/torchmetrics/text/wip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/text/wip.py -------------------------------------------------------------------------------- /src/torchmetrics/utilities/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/utilities/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/utilities/checks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/utilities/checks.py -------------------------------------------------------------------------------- /src/torchmetrics/utilities/compute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/utilities/compute.py -------------------------------------------------------------------------------- /src/torchmetrics/utilities/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/utilities/data.py -------------------------------------------------------------------------------- /src/torchmetrics/utilities/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/utilities/distributed.py -------------------------------------------------------------------------------- /src/torchmetrics/utilities/enums.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/utilities/enums.py -------------------------------------------------------------------------------- /src/torchmetrics/utilities/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/utilities/exceptions.py -------------------------------------------------------------------------------- /src/torchmetrics/utilities/imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/utilities/imports.py -------------------------------------------------------------------------------- /src/torchmetrics/utilities/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/utilities/plot.py -------------------------------------------------------------------------------- /src/torchmetrics/utilities/prints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/utilities/prints.py -------------------------------------------------------------------------------- /src/torchmetrics/video/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/video/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/video/vmaf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/video/vmaf.py -------------------------------------------------------------------------------- /src/torchmetrics/wrappers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/wrappers/__init__.py -------------------------------------------------------------------------------- /src/torchmetrics/wrappers/abstract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/wrappers/abstract.py -------------------------------------------------------------------------------- /src/torchmetrics/wrappers/bootstrapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/wrappers/bootstrapping.py -------------------------------------------------------------------------------- /src/torchmetrics/wrappers/classwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/wrappers/classwise.py -------------------------------------------------------------------------------- /src/torchmetrics/wrappers/feature_share.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/wrappers/feature_share.py -------------------------------------------------------------------------------- /src/torchmetrics/wrappers/minmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/wrappers/minmax.py -------------------------------------------------------------------------------- /src/torchmetrics/wrappers/multioutput.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/wrappers/multioutput.py -------------------------------------------------------------------------------- /src/torchmetrics/wrappers/multitask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/wrappers/multitask.py -------------------------------------------------------------------------------- /src/torchmetrics/wrappers/running.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/wrappers/running.py -------------------------------------------------------------------------------- /src/torchmetrics/wrappers/tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/wrappers/tracker.py -------------------------------------------------------------------------------- /src/torchmetrics/wrappers/transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/src/torchmetrics/wrappers/transformations.py -------------------------------------------------------------------------------- /tests/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/README.md -------------------------------------------------------------------------------- /tests/integrations/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/integrations/__init__.py -------------------------------------------------------------------------------- /tests/integrations/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/integrations/conftest.py -------------------------------------------------------------------------------- /tests/integrations/lightning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/integrations/lightning/boring_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/integrations/lightning/boring_model.py -------------------------------------------------------------------------------- /tests/integrations/test_lightning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/integrations/test_lightning.py -------------------------------------------------------------------------------- /tests/unittests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/__init__.py -------------------------------------------------------------------------------- /tests/unittests/_helpers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/_helpers/__init__.py -------------------------------------------------------------------------------- /tests/unittests/_helpers/testers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/_helpers/testers.py -------------------------------------------------------------------------------- /tests/unittests/_helpers/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/_helpers/wrappers.py -------------------------------------------------------------------------------- /tests/unittests/audio/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/audio/__init__.py -------------------------------------------------------------------------------- /tests/unittests/audio/test_c_si_snr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/audio/test_c_si_snr.py -------------------------------------------------------------------------------- /tests/unittests/audio/test_dnsmos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/audio/test_dnsmos.py -------------------------------------------------------------------------------- /tests/unittests/audio/test_nisqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/audio/test_nisqa.py -------------------------------------------------------------------------------- /tests/unittests/audio/test_pesq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/audio/test_pesq.py -------------------------------------------------------------------------------- /tests/unittests/audio/test_pit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/audio/test_pit.py -------------------------------------------------------------------------------- /tests/unittests/audio/test_sa_sdr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/audio/test_sa_sdr.py -------------------------------------------------------------------------------- /tests/unittests/audio/test_sdr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/audio/test_sdr.py -------------------------------------------------------------------------------- /tests/unittests/audio/test_si_sdr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/audio/test_si_sdr.py -------------------------------------------------------------------------------- /tests/unittests/audio/test_si_snr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/audio/test_si_snr.py -------------------------------------------------------------------------------- /tests/unittests/audio/test_snr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/audio/test_snr.py -------------------------------------------------------------------------------- /tests/unittests/audio/test_srmr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/audio/test_srmr.py -------------------------------------------------------------------------------- /tests/unittests/audio/test_stoi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/audio/test_stoi.py -------------------------------------------------------------------------------- /tests/unittests/bases/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unittests/bases/test_aggregation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/bases/test_aggregation.py -------------------------------------------------------------------------------- /tests/unittests/bases/test_collections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/bases/test_collections.py -------------------------------------------------------------------------------- /tests/unittests/bases/test_composition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/bases/test_composition.py -------------------------------------------------------------------------------- /tests/unittests/bases/test_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/bases/test_ddp.py -------------------------------------------------------------------------------- /tests/unittests/bases/test_hashing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/bases/test_hashing.py -------------------------------------------------------------------------------- /tests/unittests/bases/test_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/bases/test_metric.py -------------------------------------------------------------------------------- /tests/unittests/bases/test_saving_loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/bases/test_saving_loading.py -------------------------------------------------------------------------------- /tests/unittests/classification/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unittests/classification/_inputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/_inputs.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_accuracy.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_auc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_auc.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_auroc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_auroc.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_average_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_average_precision.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_calibration_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_calibration_error.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_cohen_kappa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_cohen_kappa.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_confusion_matrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_confusion_matrix.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_eer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_eer.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_exact_match.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_exact_match.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_f_beta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_f_beta.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_group_fairness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_group_fairness.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_hamming_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_hamming_distance.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_hinge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_hinge.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_jaccard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_jaccard.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_logauc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_logauc.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_matthews_corrcoef.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_matthews_corrcoef.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_precision_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_precision_recall.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_ranking.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_roc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_roc.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_specificity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_specificity.py -------------------------------------------------------------------------------- /tests/unittests/classification/test_stat_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/classification/test_stat_scores.py -------------------------------------------------------------------------------- /tests/unittests/clustering/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unittests/clustering/_inputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/clustering/_inputs.py -------------------------------------------------------------------------------- /tests/unittests/clustering/test_adjusted_rand_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/clustering/test_adjusted_rand_score.py -------------------------------------------------------------------------------- /tests/unittests/clustering/test_calinski_harabasz_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/clustering/test_calinski_harabasz_score.py -------------------------------------------------------------------------------- /tests/unittests/clustering/test_cluster_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/clustering/test_cluster_accuracy.py -------------------------------------------------------------------------------- /tests/unittests/clustering/test_davies_bouldin_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/clustering/test_davies_bouldin_score.py -------------------------------------------------------------------------------- /tests/unittests/clustering/test_dunn_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/clustering/test_dunn_index.py -------------------------------------------------------------------------------- /tests/unittests/clustering/test_fowlkes_mallows_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/clustering/test_fowlkes_mallows_index.py -------------------------------------------------------------------------------- /tests/unittests/clustering/test_mutual_info_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/clustering/test_mutual_info_score.py -------------------------------------------------------------------------------- /tests/unittests/clustering/test_rand_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/clustering/test_rand_score.py -------------------------------------------------------------------------------- /tests/unittests/clustering/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/clustering/test_utils.py -------------------------------------------------------------------------------- /tests/unittests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/conftest.py -------------------------------------------------------------------------------- /tests/unittests/deprecations/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unittests/deprecations/root_class_imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/deprecations/root_class_imports.py -------------------------------------------------------------------------------- /tests/unittests/detection/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/detection/__init__.py -------------------------------------------------------------------------------- /tests/unittests/detection/test_intersection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/detection/test_intersection.py -------------------------------------------------------------------------------- /tests/unittests/detection/test_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/detection/test_map.py -------------------------------------------------------------------------------- /tests/unittests/detection/test_panoptic_quality.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/detection/test_panoptic_quality.py -------------------------------------------------------------------------------- /tests/unittests/image/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/__init__.py -------------------------------------------------------------------------------- /tests/unittests/image/test_arniqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_arniqa.py -------------------------------------------------------------------------------- /tests/unittests/image/test_csi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_csi.py -------------------------------------------------------------------------------- /tests/unittests/image/test_d_lambda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_d_lambda.py -------------------------------------------------------------------------------- /tests/unittests/image/test_d_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_d_s.py -------------------------------------------------------------------------------- /tests/unittests/image/test_dists.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_dists.py -------------------------------------------------------------------------------- /tests/unittests/image/test_ergas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_ergas.py -------------------------------------------------------------------------------- /tests/unittests/image/test_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_fid.py -------------------------------------------------------------------------------- /tests/unittests/image/test_image_gradients.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_image_gradients.py -------------------------------------------------------------------------------- /tests/unittests/image/test_inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_inception.py -------------------------------------------------------------------------------- /tests/unittests/image/test_kid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_kid.py -------------------------------------------------------------------------------- /tests/unittests/image/test_lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_lpips.py -------------------------------------------------------------------------------- /tests/unittests/image/test_mifid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_mifid.py -------------------------------------------------------------------------------- /tests/unittests/image/test_ms_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_ms_ssim.py -------------------------------------------------------------------------------- /tests/unittests/image/test_perceptual_path_length.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_perceptual_path_length.py -------------------------------------------------------------------------------- /tests/unittests/image/test_psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_psnr.py -------------------------------------------------------------------------------- /tests/unittests/image/test_psnrb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_psnrb.py -------------------------------------------------------------------------------- /tests/unittests/image/test_qnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_qnr.py -------------------------------------------------------------------------------- /tests/unittests/image/test_rase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_rase.py -------------------------------------------------------------------------------- /tests/unittests/image/test_rmse_sw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_rmse_sw.py -------------------------------------------------------------------------------- /tests/unittests/image/test_sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_sam.py -------------------------------------------------------------------------------- /tests/unittests/image/test_scc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_scc.py -------------------------------------------------------------------------------- /tests/unittests/image/test_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_ssim.py -------------------------------------------------------------------------------- /tests/unittests/image/test_tv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_tv.py -------------------------------------------------------------------------------- /tests/unittests/image/test_uqi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_uqi.py -------------------------------------------------------------------------------- /tests/unittests/image/test_vif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/image/test_vif.py -------------------------------------------------------------------------------- /tests/unittests/multimodal/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unittests/multimodal/test_clip_iqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/multimodal/test_clip_iqa.py -------------------------------------------------------------------------------- /tests/unittests/multimodal/test_clip_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/multimodal/test_clip_score.py -------------------------------------------------------------------------------- /tests/unittests/multimodal/test_lve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/multimodal/test_lve.py -------------------------------------------------------------------------------- /tests/unittests/nominal/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unittests/nominal/test_cramers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/nominal/test_cramers.py -------------------------------------------------------------------------------- /tests/unittests/nominal/test_fleiss_kappa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/nominal/test_fleiss_kappa.py -------------------------------------------------------------------------------- /tests/unittests/nominal/test_pearson.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/nominal/test_pearson.py -------------------------------------------------------------------------------- /tests/unittests/nominal/test_theils_u.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/nominal/test_theils_u.py -------------------------------------------------------------------------------- /tests/unittests/nominal/test_tschuprows.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/nominal/test_tschuprows.py -------------------------------------------------------------------------------- /tests/unittests/pairwise/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unittests/pairwise/test_pairwise_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/pairwise/test_pairwise_distance.py -------------------------------------------------------------------------------- /tests/unittests/regression/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unittests/regression/test_concordance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_concordance.py -------------------------------------------------------------------------------- /tests/unittests/regression/test_cosine_similarity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_cosine_similarity.py -------------------------------------------------------------------------------- /tests/unittests/regression/test_crps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_crps.py -------------------------------------------------------------------------------- /tests/unittests/regression/test_explained_variance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_explained_variance.py -------------------------------------------------------------------------------- /tests/unittests/regression/test_js_divergence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_js_divergence.py -------------------------------------------------------------------------------- /tests/unittests/regression/test_kendall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_kendall.py -------------------------------------------------------------------------------- /tests/unittests/regression/test_kl_divergence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_kl_divergence.py -------------------------------------------------------------------------------- /tests/unittests/regression/test_log_cosh_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_log_cosh_error.py -------------------------------------------------------------------------------- /tests/unittests/regression/test_mean_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_mean_error.py -------------------------------------------------------------------------------- /tests/unittests/regression/test_minkowski_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_minkowski_distance.py -------------------------------------------------------------------------------- /tests/unittests/regression/test_pearson.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_pearson.py -------------------------------------------------------------------------------- /tests/unittests/regression/test_r2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_r2.py -------------------------------------------------------------------------------- /tests/unittests/regression/test_rse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_rse.py -------------------------------------------------------------------------------- /tests/unittests/regression/test_spearman.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_spearman.py -------------------------------------------------------------------------------- /tests/unittests/regression/test_tweedie_deviance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/regression/test_tweedie_deviance.py -------------------------------------------------------------------------------- /tests/unittests/retrieval/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unittests/retrieval/_inputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/retrieval/_inputs.py -------------------------------------------------------------------------------- /tests/unittests/retrieval/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/retrieval/helpers.py -------------------------------------------------------------------------------- /tests/unittests/retrieval/test_auroc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/retrieval/test_auroc.py -------------------------------------------------------------------------------- /tests/unittests/retrieval/test_fallout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/retrieval/test_fallout.py -------------------------------------------------------------------------------- /tests/unittests/retrieval/test_hit_rate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/retrieval/test_hit_rate.py -------------------------------------------------------------------------------- /tests/unittests/retrieval/test_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/retrieval/test_map.py -------------------------------------------------------------------------------- /tests/unittests/retrieval/test_mrr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/retrieval/test_mrr.py -------------------------------------------------------------------------------- /tests/unittests/retrieval/test_ndcg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/retrieval/test_ndcg.py -------------------------------------------------------------------------------- /tests/unittests/retrieval/test_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/retrieval/test_precision.py -------------------------------------------------------------------------------- /tests/unittests/retrieval/test_precision_recall_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/retrieval/test_precision_recall_curve.py -------------------------------------------------------------------------------- /tests/unittests/retrieval/test_r_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/retrieval/test_r_precision.py -------------------------------------------------------------------------------- /tests/unittests/retrieval/test_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/retrieval/test_recall.py -------------------------------------------------------------------------------- /tests/unittests/segmentation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/segmentation/__init__.py -------------------------------------------------------------------------------- /tests/unittests/segmentation/inputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/segmentation/inputs.py -------------------------------------------------------------------------------- /tests/unittests/segmentation/test_dice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/segmentation/test_dice.py -------------------------------------------------------------------------------- /tests/unittests/segmentation/test_hausdorff_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/segmentation/test_hausdorff_distance.py -------------------------------------------------------------------------------- /tests/unittests/segmentation/test_mean_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/segmentation/test_mean_iou.py -------------------------------------------------------------------------------- /tests/unittests/segmentation/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/segmentation/test_utils.py -------------------------------------------------------------------------------- /tests/unittests/shape/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/shape/__init__.py -------------------------------------------------------------------------------- /tests/unittests/shape/test_procrustes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/shape/test_procrustes.py -------------------------------------------------------------------------------- /tests/unittests/text/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unittests/text/_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/_helpers.py -------------------------------------------------------------------------------- /tests/unittests/text/_inputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/_inputs.py -------------------------------------------------------------------------------- /tests/unittests/text/test_bertscore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_bertscore.py -------------------------------------------------------------------------------- /tests/unittests/text/test_bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_bleu.py -------------------------------------------------------------------------------- /tests/unittests/text/test_cer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_cer.py -------------------------------------------------------------------------------- /tests/unittests/text/test_chrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_chrf.py -------------------------------------------------------------------------------- /tests/unittests/text/test_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_edit.py -------------------------------------------------------------------------------- /tests/unittests/text/test_eed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_eed.py -------------------------------------------------------------------------------- /tests/unittests/text/test_infolm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_infolm.py -------------------------------------------------------------------------------- /tests/unittests/text/test_mer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_mer.py -------------------------------------------------------------------------------- /tests/unittests/text/test_perplexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_perplexity.py -------------------------------------------------------------------------------- /tests/unittests/text/test_rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_rouge.py -------------------------------------------------------------------------------- /tests/unittests/text/test_sacre_bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_sacre_bleu.py -------------------------------------------------------------------------------- /tests/unittests/text/test_squad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_squad.py -------------------------------------------------------------------------------- /tests/unittests/text/test_ter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_ter.py -------------------------------------------------------------------------------- /tests/unittests/text/test_wer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_wer.py -------------------------------------------------------------------------------- /tests/unittests/text/test_wil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_wil.py -------------------------------------------------------------------------------- /tests/unittests/text/test_wip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/text/test_wip.py -------------------------------------------------------------------------------- /tests/unittests/utilities/test_plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/utilities/test_plot.py -------------------------------------------------------------------------------- /tests/unittests/utilities/test_utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/utilities/test_utilities.py -------------------------------------------------------------------------------- /tests/unittests/video/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/video/__init__.py -------------------------------------------------------------------------------- /tests/unittests/video/test_vmaf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/video/test_vmaf.py -------------------------------------------------------------------------------- /tests/unittests/wrappers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unittests/wrappers/test_bootstrapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/wrappers/test_bootstrapping.py -------------------------------------------------------------------------------- /tests/unittests/wrappers/test_classwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/wrappers/test_classwise.py -------------------------------------------------------------------------------- /tests/unittests/wrappers/test_feature_share.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/wrappers/test_feature_share.py -------------------------------------------------------------------------------- /tests/unittests/wrappers/test_minmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/wrappers/test_minmax.py -------------------------------------------------------------------------------- /tests/unittests/wrappers/test_multioutput.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/wrappers/test_multioutput.py -------------------------------------------------------------------------------- /tests/unittests/wrappers/test_multitask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/wrappers/test_multitask.py -------------------------------------------------------------------------------- /tests/unittests/wrappers/test_running.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/wrappers/test_running.py -------------------------------------------------------------------------------- /tests/unittests/wrappers/test_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/wrappers/test_tracker.py -------------------------------------------------------------------------------- /tests/unittests/wrappers/test_transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-AI/torchmetrics/HEAD/tests/unittests/wrappers/test_transformations.py --------------------------------------------------------------------------------