restore functional metrics (#3943)

* restore functional metrics * clean * fix
Lightning-AI · Oct 7, 2020 · b3d2526 · b3d2526
1 parent c77073f
commit b3d2526
Show file tree

Hide file tree

Showing 10 changed files with 1,974 additions and 0 deletions.
diff --git a/pytorch_lightning/metrics/functional/__init__.py b/pytorch_lightning/metrics/functional/__init__.py
@@ -0,0 +1,31 @@
+from pytorch_lightning.metrics.functional.classification import (
+    accuracy,
+    auc,
+    auroc,
+    average_precision,
+    confusion_matrix,
+    dice_score,
+    f1_score,
+    fbeta_score,
+    multiclass_precision_recall_curve,
+    multiclass_roc,
+    precision,
+    precision_recall,
+    precision_recall_curve,
+    recall,
+    roc,
+    stat_scores,
+    stat_scores_multiple_classes,
+    to_categorical,
+    to_onehot,
+    iou,
+)
+from pytorch_lightning.metrics.functional.nlp import bleu_score
+from pytorch_lightning.metrics.functional.regression import (
+    mae,
+    mse,
+    psnr,
+    rmse,
+    rmsle,
+    ssim
+)