Add Precision-Recall Curve to experimental metrics (#544)

VectorInstitute · Jan 10, 2024 · b62bc73 · b62bc73
1 parent 6965603
commit b62bc73
Show file tree

Hide file tree

Showing 7 changed files with 2,291 additions and 51 deletions.
diff --git a/cyclops/evaluate/metrics/experimental/__init__.py b/cyclops/evaluate/metrics/experimental/__init__.py
@@ -40,6 +40,11 @@
     MultilabelSensitivity,
     MultilabelTPR,
 )
+from cyclops.evaluate.metrics.experimental.precision_recall_curve import (
+    BinaryPrecisionRecallCurve,
+    MulticlassPrecisionRecallCurve,
+    MultilabelPrecisionRecallCurve,
+)
 from cyclops.evaluate.metrics.experimental.specificity import (
     BinarySpecificity,
     BinaryTNR,

diff --git a/cyclops/evaluate/metrics/experimental/functional/__init__.py b/cyclops/evaluate/metrics/experimental/functional/__init__.py
@@ -36,6 +36,11 @@
     multilabel_recall,
     multilabel_tpr,
 )
+from cyclops.evaluate.metrics.experimental.functional.precision_recall_curve import (
+    binary_precision_recall_curve,
+    multiclass_precision_recall_curve,
+    multilabel_precision_recall_curve,
+)
 from cyclops.evaluate.metrics.experimental.functional.specificity import (
     binary_specificity,
     binary_tnr,