gnn_tracking.metrics.graph_construction
#
Module Contents#
Functions#
|
Calculate efficiency and purity for edges based on |
- gnn_tracking.metrics.graph_construction.get_efficiency_purity_edges(data: torch_geometric.data.Data, pt_thld: float = 0.9, max_eta: float = 4.0) dict[str, float] #
Calculate efficiency and purity for edges based on
data.true_edge_index
.Only edges where at least one of the two nodes is accepted by the pt threshold (and reconstructable etc.) are considered.