From 95dbc1030abba2aca15a003975ec7771f6819abc Mon Sep 17 00:00:00 2001 From: Kilian Lieret Date: Mon, 7 Aug 2023 17:31:57 -0400 Subject: [PATCH] Fix: GC efficiency calculation --- src/gnn_tracking/metrics/graph_construction.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/gnn_tracking/metrics/graph_construction.py b/src/gnn_tracking/metrics/graph_construction.py index 1eb3a858..00a4018a 100644 --- 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(),