lightning/pytorch_lightning/metrics/__init__.py

from pytorch_lightning.metrics.classification import (
    Accuracy,
    AveragePrecision,
    ConfusionMatrix,
    F1,
    FBeta,
    Recall,
    ROC,
    AUROC,
    DiceCoefficient,
    MulticlassPrecisionRecallCurve,
    MulticlassROC,
    Precision,
    PrecisionRecallCurve,
    IoU,
)
from pytorch_lightning.metrics.converters import numpy_metric, tensor_metric
from pytorch_lightning.metrics.metric import Metric, TensorMetric, NumpyMetric
from pytorch_lightning.metrics.nlp import BLEUScore
from pytorch_lightning.metrics.self_supervised import EmbeddingSimilarity
from pytorch_lightning.metrics.regression import (
    MAE,
    MSE,
    PSNR,
    RMSE,
    RMSLE,
    SSIM
)
from pytorch_lightning.metrics.sklearns import (
    AUC,
    SklearnMetric,
)

__classification_metrics = [
    "AUC",
    "AUROC",
    "Accuracy",
    "AveragePrecision",
    "ConfusionMatrix",
    "DiceCoefficient",
    "F1",
    "FBeta",
    "MulticlassPrecisionRecallCurve",
    "MulticlassROC",
    "Precision",
    "PrecisionRecallCurve",
    "ROC",
    "Recall",
    "IoU",
]
__regression_metrics = [
    "MAE",
    "MSE",
    "PSNR",
    "RMSE",
    "RMSLE",
    "SSIM"
]
__sequence_metrics = ["BLEUScore"]
__selfsuper_metrics = ["EmbeddingSimilarity"]

__all__ = __regression_metrics \
    + __classification_metrics \
    + __selfsuper_metrics \
    + __sequence_metrics \
    + ["SklearnMetric"]