From 4d14cabe097a00e569efb5a19849d84f62234dea Mon Sep 17 00:00:00 2001
From: Jirka <jirka.borovec@seznam.cz>
Date: Fri, 16 Jul 2021 17:26:34 +0200
Subject: [PATCH] docs

---
 docs/source/references/functional.rst | 42 +++++++++++++--------------
 docs/source/references/modules.rst    | 28 +++++++++---------
 2 files changed, 35 insertions(+), 35 deletions(-)

diff --git a/docs/source/references/functional.rst b/docs/source/references/functional.rst
index 815eaf4ed67..28ac2b44273 100644
--- a/docs/source/references/functional.rst
+++ b/docs/source/references/functional.rst
@@ -30,6 +30,20 @@ snr [func]
     :noindex:
 
 
+psnr [func]
+~~~~~~~~~~~
+
+.. autofunction:: torchmetrics.functional.psnr
+    :noindex:
+
+
+ssim [func]
+~~~~~~~~~~~
+
+.. autofunction:: torchmetrics.functional.ssim
+    :noindex:
+
+
 **********************
 Classification Metrics
 **********************
@@ -114,10 +128,10 @@ iou [func]
 .. autofunction:: torchmetrics.functional.iou
     :noindex:
 
-kldivergence [func]
-~~~~~~~~~~~~~~~~~~~
+kl_divergence [func]
+~~~~~~~~~~~~~~~~~~~~
 
-.. autofunction:: torchmetrics.functional.kldivergence
+.. autofunction:: torchmetrics.functional.kl_divergence
     :noindex:
 
 matthews_corrcoef [func]
@@ -127,7 +141,7 @@ matthews_corrcoef [func]
     :noindex:
 
 roc [func]
-~~~~~~~~~~~~~~~~~~~~~
+~~~~~~~~~~
 
 .. autofunction:: torchmetrics.functional.roc
     :noindex:
@@ -254,17 +268,10 @@ pearson_corrcoef [func]
     :noindex:
 
 
-psnr [func]
-~~~~~~~~~~~
-
-.. autofunction:: torchmetrics.functional.psnr
-    :noindex:
-
-
-r2score [func]
-~~~~~~~~~~~~~~
+r2_score [func]
+~~~~~~~~~~~~~~~
 
-.. autofunction:: torchmetrics.functional.r2score
+.. autofunction:: torchmetrics.functional.r2_score
     :noindex:
 
 
@@ -275,13 +282,6 @@ spearman_corrcoef [func]
     :noindex:
 
 
-ssim [func]
-~~~~~~~~~~~
-
-.. autofunction:: torchmetrics.functional.ssim
-    :noindex:
-
-
 ***
 NLP
 ***
diff --git a/docs/source/references/modules.rst b/docs/source/references/modules.rst
index 3941939ae8b..100473581c2 100644
--- a/docs/source/references/modules.rst
+++ b/docs/source/references/modules.rst
@@ -322,6 +322,20 @@ KID
 .. autoclass:: torchmetrics.KID
     :noindex:
 
+
+PSNR
+~~~~
+
+.. autoclass:: torchmetrics.PSNR
+    :noindex:
+
+
+SSIM
+~~~~
+
+.. autoclass:: torchmetrics.SSIM
+    :noindex:
+
 ******************
 Regression Metrics
 ******************
@@ -375,13 +389,6 @@ PearsonCorrcoef
     :noindex:
 
 
-PSNR
-~~~~
-
-.. autoclass:: torchmetrics.PSNR
-    :noindex:
-
-
 R2Score
 ~~~~~~~
 
@@ -396,13 +403,6 @@ SpearmanCorrcoef
     :noindex:
 
 
-SSIM
-~~~~
-
-.. autoclass:: torchmetrics.SSIM
-    :noindex:
-
-
 
 *********
 Retrieval