Port plot_confusion_matrix to new API.

SpikeInterface · Sep 19, 2023 · e49071e · e49071e
1 parent 46c4ada
commit e49071e
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 95 deletions.
diff --git a/src/spikeinterface/widgets/_legacy_mpl_widgets/__init__.py b/src/spikeinterface/widgets/_legacy_mpl_widgets/__init__.py
@@ -15,9 +15,6 @@
 # units on probe
 from .unitprobemap import plot_unit_probe_map, UnitProbeMapWidget
 
-# comparison related
-from .confusionmatrix import plot_confusion_matrix, ConfusionMatrixWidget
-
 from .multicompgraph import (
     plot_multicomp_graph,
     MultiCompGraphWidget,

diff --git a/src/spikeinterface/widgets/_legacy_mpl_widgets/confusionmatrix.py b/src/spikeinterface/widgets/_legacy_mpl_widgets/confusionmatrix.py
diff --git a/src/spikeinterface/widgets/tests/test_widgets.py b/src/spikeinterface/widgets/tests/test_widgets.py
@@ -330,6 +330,12 @@ def test_plot_agreement_matrix(self):
             if backend not in self.skip_backends:
                 sw.plot_agreement_matrix(self.gt_comp)
 
+    def test_plot_confusion_matrix(self):
+        possible_backends = list(sw.AgreementMatrixWidget.get_possible_backends())
+        for backend in possible_backends:
+            if backend not in self.skip_backends:
+                sw.plot_confusion_matrix(self.gt_comp)
+
 
 
 if __name__ == "__main__":
@@ -352,7 +358,8 @@ def test_plot_agreement_matrix(self):
     # mytest.test_quality_metrics()
     # mytest.test_template_metrics()
     # mytest.test_amplitudes()
-    mytest.test_plot_agreement_matrix()
+    # mytest.test_plot_agreement_matrix()
+    mytest.test_plot_confusion_matrix()
 
     # plt.ion()
     plt.show()
diff --git a/src/spikeinterface/widgets/widget_list.py b/src/spikeinterface/widgets/widget_list.py
@@ -6,6 +6,7 @@
 from .all_amplitudes_distributions import AllAmplitudesDistributionsWidget
 from .amplitudes import AmplitudesWidget
 from .autocorrelograms import AutoCorrelogramsWidget
+from .confusion_matrix import ConfusionMatrixWidget
 from .crosscorrelograms import CrossCorrelogramsWidget
 from .motion import MotionWidget
 from .quality_metrics import QualityMetricsWidget
@@ -28,6 +29,7 @@
     AllAmplitudesDistributionsWidget,
     AmplitudesWidget,
     AutoCorrelogramsWidget,
+    ConfusionMatrixWidget,
     CrossCorrelogramsWidget,
     MotionWidget,
     QualityMetricsWidget,
@@ -82,6 +84,7 @@
 plot_all_amplitudes_distributions = AllAmplitudesDistributionsWidget
 plot_amplitudes = AmplitudesWidget
 plot_autocorrelograms = AutoCorrelogramsWidget
+plot_confusion_matrix = ConfusionMatrixWidget
 plot_crosscorrelograms = CrossCorrelogramsWidget
 plot_motion = MotionWidget
 plot_quality_metrics = QualityMetricsWidget