:py:mod:`gnn_tracking.metrics.graph_construction`
=================================================

.. py:module:: gnn_tracking.metrics.graph_construction


Module Contents
---------------


Functions
~~~~~~~~~

.. autoapisummary::

   gnn_tracking.metrics.graph_construction.get_efficiency_purity_edges



.. py:function:: 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.