mlreco.post_processing.metrics.cluster_gnn_metrics module

mlreco.post_processing.metrics.cluster_gnn_metrics.cluster_gnn_metrics(cfg, module_cfg, data_blob, res, logdir, iteration, edge_pred=None, clusts=None, node_pred=None, edge_index=None, clust_data=None, particles=None, data_idx=None, clust_data_noghost=None, **kwargs)[source]

Compute metrics for GRAPPA stage (GNN clustering).

enable_physics_metrics = compute detailed cluster-wise metrics integrated_metrics = compute voxel-wise ARI/Purity/Efficiency metrics (only if enable_physics_metrics: False)

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.cluster_cnn_metrics module mlreco.post_processing.metrics.cosmic_discriminator_metrics module