Fix: GC efficiency calculation

gnn-tracking · Aug 7, 2023 · 95dbc10 · 95dbc10
1 parent cad7366
commit 95dbc10
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gnn_tracking/metrics/graph_construction.py b/src/gnn_tracking/metrics/graph_construction.py
@@ -13,7 +13,7 @@ def get_efficiency_purity_edges(data: Data, pt_thld=0.9) -> dict[str, float]:
         hit_mask[data.true_edge_index[0]] & hit_mask[data.true_edge_index[1]]
     )
     # Factor of 2 because the true edges are undirected
-    efficiency = data.y[edge_mask].sum() / 2 * true_edge_mask.sum()
+    efficiency = data.y[edge_mask].sum() / (2 * true_edge_mask.sum())
     purity = data.y[edge_mask].sum() / edge_mask.sum()
     return {
         "efficiency": efficiency.item(),