fixbug: use BinaryPrecisionRecallCurve instead of PrecisionRecallCurve (

#1956)
openvinotoolkit · Apr 9, 2024 · 7a963e9 · 7a963e9
1 parent 25fc952
commit 7a963e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/anomalib/metrics/aupr.py b/src/anomalib/metrics/aupr.py
@@ -6,14 +6,14 @@
 
 import torch
 from matplotlib.figure import Figure
-from torchmetrics import PrecisionRecallCurve
+from torchmetrics.classification import BinaryPrecisionRecallCurve
 from torchmetrics.utilities.compute import auc
 from torchmetrics.utilities.data import dim_zero_cat
 
 from .plotting_utils import plot_figure
 
 
-class AUPR(PrecisionRecallCurve):
+class AUPR(BinaryPrecisionRecallCurve):
     """Area under the PR curve.
 
     This metric computes the area under the precision-recall curve.