diff --git a/torcheval/metrics/__init__.py b/torcheval/metrics/__init__.py index d404871b..4b0e7665 100644 --- a/torcheval/metrics/__init__.py +++ b/torcheval/metrics/__init__.py @@ -72,6 +72,7 @@ "BinaryAccuracy", "BinaryAUPRC", "BinaryAUROC", + "BinaryBinnedAUPRC", "BinaryBinnedAUROC", "BinaryBinnedPrecisionRecallCurve", "BinaryConfusionMatrix",