__all__

Lightning-AI · May 25, 2020 · c9fcdf9 · c9fcdf9
1 parent 0c80f56
commit c9fcdf9
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,