__all__

Lightning-AI · Apr 27, 2020 · 667aec2 · 667aec2
1 parent e440404
commit 667aec2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pytorch_lightning/metrics/sklearn.py b/pytorch_lightning/metrics/sklearn.py
@@ -6,6 +6,9 @@
 from pytorch_lightning import _logger as lightning_logger
 from pytorch_lightning.metrics.metric import NumpyMetric
 
+__all__ = ['SklearnMetric', 'Accuracy', 'AveragePrecision', 'AUC', 'ConfusionMatrix', 'F1', 'FBeta',
+           'Precision', 'Recall', 'PrecisionRecallCurve', 'ROC', 'AUROC']
+
 
 class SklearnMetric(NumpyMetric):
     def __init__(self, metric_name: str,