Annotating napari classes

alan-turing-institute · Aug 30, 2023 · 2bed123 · 2bed123
1 parent 9fc0321
commit 2bed123
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/grace/napari/utils.py b/grace/napari/utils.py
@@ -67,7 +67,7 @@ def cut_graph_using_mask(
     mask : array
         A binary mask to filter points and edges in the graph.
     update_graph : bool, (default: True)
-        Update edge attributes in place.
+        Update node & edge attributes in place.
 
     Returns
     -------

diff --git a/grace/utils/metrics.py b/grace/utils/metrics.py
@@ -33,8 +33,8 @@ def confusion_matrix_metric(
     edge_pred: torch.Tensor,
     node_true: torch.Tensor,
     edge_true: torch.Tensor,
-    node_classes: List[str] = ["TP", "TN"],
-    edge_classes: List[str] = ["TP", "TN"],
+    node_classes: List[str] = ["TN_node", "TP_node"],
+    edge_classes: List[str] = ["TN_edge", "TP_edge"],
     figsize: Tuple[int] = (6, 5),
 ) -> Tuple[plt.Figure]:
     node_pred_labels = node_pred.argmax(dim=-1)