mlreco.post_processing.metrics.kinematics_metrics module

mlreco.post_processing.metrics.kinematics_metrics.kinematics_metrics(cfg, module_cfg, data_blob, res, logdir, iteration, data_idx=None, clust_data=None, particles=None, kinematics=None, particle_graph=None, kinematics_particles=None, inter_particles=None, node_pred_type=None, node_pred_p=None, flow_edge_pred=None, kinematics_edge_index=None, counter=None, **kwargs)[source]

Compute metrics for particle hierarchy + kinematics stage (GNN).

Parameters

  • data_blob (dict) – The input data dictionary from iotools.

  • res (dict) – The output of the network, formatted using analysis_keys.

  • cfg (dict) – Configuration.

  • logdir (string) – Path to folder where CSV logs can be stored.

  • iteration (int) – Current iteration number.

Notes

N/A.

mlreco.post_processing.metrics.graph_spice_metrics module mlreco.post_processing.metrics.pid_metrics module