Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | V | Z _ __call__() (gnn_tracking.graph_construction.k_scanner.GraphConstructionKNNScanner method) (gnn_tracking.metrics.cluster_metrics.ClusterMetricType method) (gnn_tracking.metrics.losses.LossFctType method) (gnn_tracking.postprocessing.clusterscanner.ClusterScanner method) (gnn_tracking.postprocessing.clusterscanner.CombinedClusterScanner method) (gnn_tracking.postprocessing.dbscanscanner.DBSCANHyperParamScanner method) (gnn_tracking.postprocessing.dbscanscanner.DBSCANPerformanceDetails method) (gnn_tracking.utils.timing.Timer method) __repr__() (gnn_tracking.models.dynamic_edge_conv.DynamicEdgeConv method) __version__ (in module gnn_tracking) _background_loss() (in module gnn_tracking.metrics.losses) _binary_focal_loss() (in module gnn_tracking.metrics.losses) _build_graph() (gnn_tracking.metrics.losses.GraphConstructionHingeEmbeddingLoss method) _condensation_loss() (in module gnn_tracking.metrics.losses) _DEFAULT_FEATURE_SCALE (in module gnn_tracking.preprocessing.point_cloud_builder) _DEFAULT_KS (in module gnn_tracking.graph_construction.k_scanner) _eval_spline() (gnn_tracking.graph_construction.k_scanner.KScanResults method) _evaluate_cluster_metrics() (gnn_tracking.training.tc.TCModule method) _evaluate_graph() (gnn_tracking.graph_construction.k_scanner.GraphConstructionKNNScanner method) _extra_metrics (gnn_tracking.graph_construction.k_scanner.KScanResults attribute) _extract_all_rotations() (gnn_tracking.preprocessing.exatrkx_cell_features.DetectorRotations method) _extract_all_size() (gnn_tracking.preprocessing.exatrkx_cell_features.DetectorPixelSize method) _extract_all_thicknesses() (gnn_tracking.preprocessing.exatrkx_cell_features.DetectorThicknesses method) _extract_rotation_matrix() (gnn_tracking.preprocessing.exatrkx_cell_features.DetectorRotations method) _false() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) _fast_condensation_loss() (in module gnn_tracking.metrics.losses) _find_loggers() (gnn_tracking.training.callbacks.ExpandWandbConfig method) _first_occurrences() (in module gnn_tracking.metrics.losses) _fix_datatypes() (gnn_tracking.utils.loading.TrackingDataModule static method) _forward() (gnn_tracking.metrics.losses.EdgeWeightBCELoss static method) (gnn_tracking.metrics.losses.EdgeWeightFocalLoss method) (gnn_tracking.metrics.losses.FalsifyLowPtEdgeWeightLoss method) (gnn_tracking.models.resin.ResidualNetwork method) (gnn_tracking.models.resin.Skip1ResidualNetwork method) (gnn_tracking.models.resin.Skip2ResidualNetwork method) (gnn_tracking.models.resin.SkipTopResidualNetwork method) _get_best_trials() (gnn_tracking.postprocessing.dbscanscanner.DBSCANHyperParamScanner method) _get_config() (gnn_tracking.training.callbacks.ExpandWandbConfig method) _get_dataloader() (gnn_tracking.utils.loading.TrackingDataModule method) _get_dataset() (gnn_tracking.utils.loading.TrackingDataModule method) _get_edge_index() (gnn_tracking.preprocessing.point_cloud_builder.PointCloudBuilder method) _get_foms_at_target() (gnn_tracking.graph_construction.k_scanner.KScanResults method) _get_paths() (gnn_tracking.utils.loading.TrackingDataset static method) _get_target_k() (gnn_tracking.graph_construction.k_scanner.KScanResults method) _hinge_loss_components() (in module gnn_tracking.metrics.losses) _init_rotation_array() (gnn_tracking.preprocessing.exatrkx_cell_features.DetectorRotations method) _init_size_array() (gnn_tracking.preprocessing.exatrkx_cell_features.DetectorPixelSize method) _init_thickness_array() (gnn_tracking.preprocessing.exatrkx_cell_features.DetectorThicknesses method) _log_errors() (gnn_tracking.training.base.ImprovedLogLM method) _mse() (gnn_tracking.metrics.losses.ObjectLoss static method) _P (in module gnn_tracking.utils.dictionaries) _predicted_false() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) _predicted_true() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) _reset_graph() (gnn_tracking.postprocessing.fastrescanner.DBSCANFastRescan method) _reset_layer_parameters() (gnn_tracking.models.edge_filter.EFMLP static method) (gnn_tracking.models.graph_construction.GraphConstructionFCNN static method) _reset_trials() (gnn_tracking.postprocessing.dbscanscanner.DBSCANHyperParamScanner method) (gnn_tracking.postprocessing.dbscanscanner.DBSCANHyperParamScannerFixed method) _save_hparams() (gnn_tracking.graph_construction.data_transformer.DataTransformer method) _sklearn_signature_wrap() (in module gnn_tracking.metrics.cluster_metrics) _spline() (gnn_tracking.graph_construction.k_scanner.KScanResults method) _sqconvex_combination() (in module gnn_tracking.models.resin) _square_distances() (in module gnn_tracking.metrics.losses) _tracking_metrics_nan_results (in module gnn_tracking.metrics.cluster_metrics) _true() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) _validate_config() (gnn_tracking.models.graph_construction.MLGraphConstruction method) _validate_settings() (gnn_tracking.training.tc.TCModule method) A acc() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) add_key_prefix() (in module gnn_tracking.utils.dictionaries) add_key_suffix() (in module gnn_tracking.utils.dictionaries) add_legend() (gnn_tracking.analysis.edge_classification.ThresholdTrackInfoPlot method) append_coordinates() (gnn_tracking.utils.plotting.EventPlotter method) append_features() (gnn_tracking.preprocessing.point_cloud_builder.PointCloudBuilder method) args (in module gnn_tracking.graph_construction.build_graphs) (in module gnn_tracking.preprocessing.build_point_clouds) assert_feat_dim() (in module gnn_tracking.utils.asserts) assert_version_geq() (in module gnn_tracking.utils.versioning) augment_hit_features() (in module gnn_tracking.preprocessing.exatrkx_cell_features) B BackgroundLoss (class in gnn_tracking.metrics.losses) backward() (gnn_tracking.training.base.TrackingModule method) balanced_acc() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) binary_focal_loss() (in module gnn_tracking.metrics.losses) BinaryClassificationStats (class in gnn_tracking.metrics.binary_classification) build_edges() (gnn_tracking.graph_construction.graph_builder.GraphBuilder method) C calc_dphi() (gnn_tracking.graph_construction.graph_builder.GraphBuilder method) calc_eta() (gnn_tracking.graph_construction.graph_builder.GraphBuilder method) (gnn_tracking.preprocessing.point_cloud_builder.PointCloudBuilder method) (gnn_tracking.utils.plotting.EventPlotter method) cartesian_to_spherical() (in module gnn_tracking.preprocessing.exatrkx_cell_features) cf_logger (in module gnn_tracking.preprocessing.exatrkx_cell_features) cli_main() (in module gnn_tracking.training.run) cluster() (gnn_tracking.postprocessing.fastrescanner.DBSCANFastRescan method) ClusterMetricType (class in gnn_tracking.metrics.cluster_metrics) ClusterScanner (class in gnn_tracking.postprocessing.clusterscanner) collect_all_ec_stats() (in module gnn_tracking.analysis.edge_classification) CombinedClusterScanner (class in gnn_tracking.postprocessing.clusterscanner) common_metrics (in module gnn_tracking.metrics.cluster_metrics) compute() (gnn_tracking.utils.lightning.StandardError method) concat_edge_embeddings_length (gnn_tracking.models.resin.ResIN property) configure_optimizers() (gnn_tracking.training.base.TrackingModule method) configure_plt() (gnn_tracking.utils.plotting.GraphPlotter method) correct_truth_labels() (gnn_tracking.graph_construction.graph_builder.GraphBuilder method) count_hits_per_cluster() (in module gnn_tracking.metrics.cluster_metrics) count_tracking_metrics() (in module gnn_tracking.metrics.cluster_metrics) D data_list (gnn_tracking.graph_construction.graph_builder.GraphBuilder property) data_preproc() (gnn_tracking.training.base.TrackingModule method) datasets (gnn_tracking.utils.loading.TrackingDataModule property) DataTransformer (class in gnn_tracking.graph_construction.data_transformer) dbscan() (in module gnn_tracking.postprocessing.dbscanscanner) DBSCANFastRescan (class in gnn_tracking.postprocessing.fastrescanner) DBSCANHyperParamScanner (class in gnn_tracking.postprocessing.dbscanscanner) DBSCANHyperParamScannerFixed (class in gnn_tracking.postprocessing.dbscanscanner) DBSCANPerformanceDetails (class in gnn_tracking.postprocessing.dbscanscanner) DEFAULT_FEATURES (in module gnn_tracking.preprocessing.point_cloud_builder) denote_pt() (in module gnn_tracking.utils.nomenclature) DetectorPixelSize (class in gnn_tracking.preprocessing.exatrkx_cell_features) DetectorRotations (class in gnn_tracking.preprocessing.exatrkx_cell_features) DetectorThicknesses (class in gnn_tracking.preprocessing.exatrkx_cell_features) determine_array_size() (in module gnn_tracking.preprocessing.exatrkx_cell_features) df (gnn_tracking.postprocessing.dbscanscanner.OCScanResults property) df_mean (gnn_tracking.postprocessing.dbscanscanner.OCScanResults property) disable() (gnn_tracking.utils.lightning.SimpleTqdmProgressBar method) distance_largest_segments (gnn_tracking.analysis.graphs.TrackGraphInfo attribute), [1] double_majority (gnn_tracking.metrics.cluster_metrics.TrackingMetrics attribute) DynamicEdgeConv (class in gnn_tracking.models.dynamic_edge_conv) E ECCut (class in gnn_tracking.graph_construction.data_transformer) ECCutRefine (class in gnn_tracking.graph_construction.data_transformer) ECForGraphTCN (class in gnn_tracking.models.edge_classifier) ECFromChkpt (class in gnn_tracking.models.edge_classifier) ECModule (class in gnn_tracking.training.ec) EdgeWeightBCELoss (class in gnn_tracking.metrics.losses) EdgeWeightFocalLoss (class in gnn_tracking.metrics.losses) EFDeepSet (class in gnn_tracking.models.edge_filter) EFMLP (class in gnn_tracking.models.edge_filter) EventPlotter (class in gnn_tracking.utils.plotting) expand_grid() (in module gnn_tracking.utils.dictionaries) ExpandWandbConfig (class in gnn_tracking.training.callbacks) extract_dir_new() (in module gnn_tracking.preprocessing.exatrkx_cell_features) F F1() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) fake_double_majority (gnn_tracking.metrics.cluster_metrics.TrackingMetrics attribute) fake_lhc (gnn_tracking.metrics.cluster_metrics.TrackingMetrics attribute) fake_perfect (gnn_tracking.metrics.cluster_metrics.TrackingMetrics attribute) falsify_low_pt_edges() (in module gnn_tracking.metrics.losses) FalsifyLowPtEdgeWeightLoss (class in gnn_tracking.metrics.losses) find_latest_checkpoint() (in module gnn_tracking.utils.lightning) fix_seeds() (in module gnn_tracking.utils.seeds) flatten_track_metrics() (in module gnn_tracking.metrics.cluster_metrics) FN() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) FNR() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) format_results_table() (in module gnn_tracking.training.callbacks) forward() (gnn_tracking.graph_construction.data_transformer.ECCut method) (gnn_tracking.graph_construction.data_transformer.ECCutRefine method) (gnn_tracking.metrics.losses.BackgroundLoss method) (gnn_tracking.metrics.losses.FalsifyLowPtEdgeWeightLoss method) (gnn_tracking.metrics.losses.GraphConstructionHingeEmbeddingLoss method) (gnn_tracking.metrics.losses.HaughtyFocalLoss method) (gnn_tracking.metrics.losses.LossClones method) (gnn_tracking.metrics.losses.ObjectLoss method) (gnn_tracking.metrics.losses.PotentialLoss method) (gnn_tracking.models.dynamic_edge_conv.DynamicEdgeConv method) (gnn_tracking.models.edge_classifier.ECForGraphTCN method) (gnn_tracking.models.edge_classifier.PerfectEdgeClassification method) (gnn_tracking.models.edge_filter.EFDeepSet method) (gnn_tracking.models.edge_filter.EFMLP method) (gnn_tracking.models.edge_filter.GeometricEF method) (gnn_tracking.models.graph_construction.GraphConstructionFCNN method) (gnn_tracking.models.graph_construction.GraphConstructionResIN method) (gnn_tracking.models.graph_construction.MLGraphConstruction method) (gnn_tracking.models.graph_construction.MLPCTransformer method) (gnn_tracking.models.interaction_network.InteractionNetwork method) (gnn_tracking.models.meta.Sequential method) (gnn_tracking.models.mlp.MLP method) (gnn_tracking.models.resin.ResidualNetwork method) (gnn_tracking.models.resin.ResIN method) (gnn_tracking.models.track_condensation_networks.GraphTCN method) (gnn_tracking.models.track_condensation_networks.INConvBlock method) (gnn_tracking.models.track_condensation_networks.ModularGraphTCN method) (gnn_tracking.models.track_condensation_networks.PerfectECGraphTCN method) (gnn_tracking.models.track_condensation_networks.PointCloudTCN method) (gnn_tracking.models.track_condensation_networks.PreTrainedECGraphTCN method) (gnn_tracking.training.base.TrackingModule method) FP() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) FPR() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) freeze() (in module gnn_tracking.utils.torch_utils) freeze_if() (in module gnn_tracking.utils.torch_utils) from_chkpt() (gnn_tracking.models.graph_construction.MLGraphConstruction class method) from_ml_chkpt() (gnn_tracking.models.graph_construction.MLPCTransformer class method) G GeometricEF (class in gnn_tracking.models.edge_filter) get() (gnn_tracking.utils.loading.TrackingDataset method) get_all() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) get_all_argument_names() (in module gnn_tracking.utils.signature) get_all_ec_stats() (in module gnn_tracking.analysis.edge_classification) get_all_graph_construction_stats() (in module gnn_tracking.analysis.graphs) get_all_local_angles() (in module gnn_tracking.preprocessing.exatrkx_cell_features) get_all_rotated() (in module gnn_tracking.preprocessing.exatrkx_cell_features) get_basic_counts() (in module gnn_tracking.analysis.graphs) get_cc_labels() (in module gnn_tracking.analysis.graphs) get_collateral_mask() (gnn_tracking.analysis.latent.SelectedPidsPlot method) get_color_mapper() (in module gnn_tracking.analysis.latent) get_colors() (gnn_tracking.analysis.latent.SelectedPidsPlot method) get_commit_hash() (in module gnn_tracking.utils.versioning) get_dataframe() (gnn_tracking.graph_construction.graph_builder.GraphBuilder method) get_edge_index() (gnn_tracking.models.dynamic_edge_conv.DynamicEdgeConv method) get_edge_mask_from_node_mask() (in module gnn_tracking.utils.graph_masks) get_efficiency_purity_edges() (in module gnn_tracking.metrics.graph_construction) get_event_id_sector_from_str() (gnn_tracking.graph_construction.graph_builder.GraphBuilder static method) get_foms() (gnn_tracking.graph_construction.k_scanner.GraphConstructionKNNScanner method) (gnn_tracking.graph_construction.k_scanner.KScanResults method) (gnn_tracking.postprocessing.clusterscanner.ClusterScanner method) (gnn_tracking.postprocessing.clusterscanner.CombinedClusterScanner method) (gnn_tracking.postprocessing.dbscanscanner.DBSCANHyperParamScanner method) (gnn_tracking.postprocessing.dbscanscanner.DBSCANPerformanceDetails method) (gnn_tracking.postprocessing.dbscanscanner.OCScanResults method) get_good_node_mask() (in module gnn_tracking.utils.graph_masks) get_hits() (gnn_tracking.utils.plotting.EventPlotter method) get_largest_segment_fracs() (in module gnn_tracking.analysis.graphs) get_lightning_module() (in module gnn_tracking.utils.lightning) get_logger() (in module gnn_tracking.utils.log) get_losses() (gnn_tracking.training.ec.ECModule method) (gnn_tracking.training.ml.MLModule method) (gnn_tracking.training.tc.TCModule method) get_maximized_bcs() (in module gnn_tracking.metrics.binary_classification) get_measurements() (gnn_tracking.graph_construction.graph_builder.GraphBuilder method) (gnn_tracking.preprocessing.point_cloud_builder.PointCloudBuilder method) get_model() (in module gnn_tracking.utils.lightning) get_n_best_trials() (gnn_tracking.postprocessing.dbscanscanner.OCScanResults method) get_n_reachable() (in module gnn_tracking.analysis.graphs) get_n_truth_edges() (gnn_tracking.graph_construction.graph_builder.GraphBuilder method) get_object_from_path() (in module gnn_tracking.utils.lightning) get_orphan_counts() (in module gnn_tracking.analysis.graphs) get_parser() (in module gnn_tracking.graph_construction.build_graphs) (in module gnn_tracking.preprocessing.build_point_clouds) get_pixel_size() (gnn_tracking.preprocessing.exatrkx_cell_features.DetectorPixelSize method) get_results() (gnn_tracking.graph_construction.k_scanner.GraphConstructionKNNScanner method) (gnn_tracking.postprocessing.dbscanscanner.DBSCANHyperParamScanner method) (gnn_tracking.postprocessing.dbscanscanner.DBSCANPerformanceDetails method) get_roc_auc_scores() (in module gnn_tracking.metrics.binary_classification) get_rotations() (gnn_tracking.preprocessing.exatrkx_cell_features.DetectorRotations method) get_thicknesses() (gnn_tracking.preprocessing.exatrkx_cell_features.DetectorThicknesses method) get_track_graph_info() (in module gnn_tracking.analysis.graphs) get_track_graph_info_from_data() (in module gnn_tracking.analysis.graphs) get_truth_edge_index() (in module gnn_tracking.preprocessing.point_cloud_builder) get_two_hop_tuples() (in module gnn_tracking.graph_construction.graph_builder) gnn_tracking module gnn_tracking.analysis module gnn_tracking.analysis.edge_classification module gnn_tracking.analysis.graphs module gnn_tracking.analysis.latent module gnn_tracking.graph_construction module gnn_tracking.graph_construction.build_graphs module gnn_tracking.graph_construction.build_graphs_hpo module gnn_tracking.graph_construction.build_module_map module gnn_tracking.graph_construction.data_transformer module gnn_tracking.graph_construction.graph_builder module gnn_tracking.graph_construction.k_scanner module gnn_tracking.metrics module gnn_tracking.metrics.binary_classification module gnn_tracking.metrics.cluster_metrics module gnn_tracking.metrics.graph_construction module gnn_tracking.metrics.losses module gnn_tracking.models module gnn_tracking.models.dynamic_edge_conv module gnn_tracking.models.edge_classifier module gnn_tracking.models.edge_filter module gnn_tracking.models.graph_construction module gnn_tracking.models.interaction_network module gnn_tracking.models.meta module gnn_tracking.models.mlp module gnn_tracking.models.resin module gnn_tracking.models.track_condensation_networks module gnn_tracking.postprocessing module gnn_tracking.postprocessing.clusterscanner module gnn_tracking.postprocessing.dbscanscanner module gnn_tracking.postprocessing.fastrescanner module gnn_tracking.preprocessing module gnn_tracking.preprocessing.build_point_clouds module gnn_tracking.preprocessing.exatrkx_cell_features module gnn_tracking.preprocessing.point_cloud_builder module gnn_tracking.training module gnn_tracking.training.base module gnn_tracking.training.callbacks module gnn_tracking.training.ec module gnn_tracking.training.ml module gnn_tracking.training.run module gnn_tracking.training.tc module gnn_tracking.utils module gnn_tracking.utils.asserts module gnn_tracking.utils.colors module gnn_tracking.utils.device module gnn_tracking.utils.dictionaries module gnn_tracking.utils.graph_masks module gnn_tracking.utils.lightning module gnn_tracking.utils.loading module gnn_tracking.utils.log module gnn_tracking.utils.math module gnn_tracking.utils.nomenclature module gnn_tracking.utils.oom module gnn_tracking.utils.plotting module gnn_tracking.utils.seeds module gnn_tracking.utils.signature module gnn_tracking.utils.timing module gnn_tracking.utils.torch_utils module gnn_tracking.utils.versioning module GRAPH_INDIR (in module gnn_tracking.graph_construction.build_graphs_hpo) GraphBuilder (class in gnn_tracking.graph_construction.graph_builder) GraphConstructionFCNN (class in gnn_tracking.models.graph_construction) GraphConstructionHingeEmbeddingLoss (class in gnn_tracking.metrics.losses) GraphConstructionKNNScanner (class in gnn_tracking.graph_construction.k_scanner) GraphConstructionResIN (class in gnn_tracking.models.graph_construction) GraphPlotter (class in gnn_tracking.utils.plotting) GraphTCN (class in gnn_tracking.models.track_condensation_networks) guess_device() (in module gnn_tracking.utils.device) H HaughtyFocalLoss (class in gnn_tracking.metrics.losses) highlight_metric() (gnn_tracking.training.ec.ECModule method) (gnn_tracking.training.ml.MLModule method) (gnn_tracking.training.tc.TCModule method) hits_per_cluster_count_to_flat_dict() (in module gnn_tracking.metrics.cluster_metrics) I ImprovedLogLM (class in gnn_tracking.training.base) INConvBlock (class in gnn_tracking.models.track_condensation_networks) InteractionNetwork (class in gnn_tracking.models.interaction_network) is_enabled (gnn_tracking.utils.lightning.SimpleTqdmProgressBar property) is_last_val_batch() (gnn_tracking.training.tc.TCModule method) is_oom_error() (in module gnn_tracking.utils.oom) K knn (in module gnn_tracking.models.dynamic_edge_conv) knn_with_max_radius() (in module gnn_tracking.models.graph_construction) KScanResults (class in gnn_tracking.graph_construction.k_scanner) L len() (gnn_tracking.utils.loading.TrackingDataset method) lhc (gnn_tracking.metrics.cluster_metrics.TrackingMetrics attribute) lighten_color() (in module gnn_tracking.utils.colors) load_detector() (in module gnn_tracking.preprocessing.exatrkx_cell_features) load_obj_from_hparams() (in module gnn_tracking.utils.lightning) LOG_DEFAULT_LEVEL (in module gnn_tracking.utils.log) log_dict_with_errors() (gnn_tracking.training.base.ImprovedLogLM method) logger (in module gnn_tracking.utils.log) loss_weight_type (in module gnn_tracking.metrics.losses) LossClones (class in gnn_tracking.metrics.losses) LossFctType (class in gnn_tracking.metrics.losses) M MCC() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) MEASUREMENT_OUTDIR (in module gnn_tracking.graph_construction.build_graphs_hpo) message() (gnn_tracking.models.dynamic_edge_conv.DynamicEdgeConv method) (gnn_tracking.models.interaction_network.InteractionNetwork method) MLGraphConstruction (class in gnn_tracking.models.graph_construction) MLGraphConstructionFromChkpt (class in gnn_tracking.models.graph_construction) MLModule (class in gnn_tracking.training.ml) MLP (class in gnn_tracking.models.mlp) MLPCTransformer (class in gnn_tracking.models.graph_construction) MLPCTransformerFromMLChkpt (class in gnn_tracking.models.graph_construction) ModularGraphTCN (class in gnn_tracking.models.track_condensation_networks) module gnn_tracking gnn_tracking.analysis gnn_tracking.analysis.edge_classification gnn_tracking.analysis.graphs gnn_tracking.analysis.latent gnn_tracking.graph_construction gnn_tracking.graph_construction.build_graphs gnn_tracking.graph_construction.build_graphs_hpo gnn_tracking.graph_construction.build_module_map gnn_tracking.graph_construction.data_transformer gnn_tracking.graph_construction.graph_builder gnn_tracking.graph_construction.k_scanner gnn_tracking.metrics gnn_tracking.metrics.binary_classification gnn_tracking.metrics.cluster_metrics gnn_tracking.metrics.graph_construction gnn_tracking.metrics.losses gnn_tracking.models gnn_tracking.models.dynamic_edge_conv gnn_tracking.models.edge_classifier gnn_tracking.models.edge_filter gnn_tracking.models.graph_construction gnn_tracking.models.interaction_network gnn_tracking.models.meta gnn_tracking.models.mlp gnn_tracking.models.resin gnn_tracking.models.track_condensation_networks gnn_tracking.postprocessing gnn_tracking.postprocessing.clusterscanner gnn_tracking.postprocessing.dbscanscanner gnn_tracking.postprocessing.fastrescanner gnn_tracking.preprocessing gnn_tracking.preprocessing.build_point_clouds gnn_tracking.preprocessing.exatrkx_cell_features gnn_tracking.preprocessing.point_cloud_builder gnn_tracking.training gnn_tracking.training.base gnn_tracking.training.callbacks gnn_tracking.training.ec gnn_tracking.training.ml gnn_tracking.training.run gnn_tracking.training.tc gnn_tracking.utils gnn_tracking.utils.asserts gnn_tracking.utils.colors gnn_tracking.utils.device gnn_tracking.utils.dictionaries gnn_tracking.utils.graph_masks gnn_tracking.utils.lightning gnn_tracking.utils.loading gnn_tracking.utils.log gnn_tracking.utils.math gnn_tracking.utils.nomenclature gnn_tracking.utils.oom gnn_tracking.utils.plotting gnn_tracking.utils.seeds gnn_tracking.utils.signature gnn_tracking.utils.timing gnn_tracking.utils.torch_utils gnn_tracking.utils.versioning N n_cleaned_clusters (gnn_tracking.metrics.cluster_metrics.TrackingMetrics attribute) N_EXPERIMENTS (in module gnn_tracking.graph_construction.build_graphs_hpo) n_false() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) N_GRAPHS_MEASURED (in module gnn_tracking.graph_construction.build_graphs_hpo) n_hits (gnn_tracking.analysis.graphs.TrackGraphInfo attribute), [1] n_hits_largest_component (gnn_tracking.analysis.graphs.TrackGraphInfo attribute), [1] n_hits_largest_segment (gnn_tracking.analysis.graphs.TrackGraphInfo attribute), [1] N_OOM_ERRORS (in module gnn_tracking.utils.oom) n_orphan_correct (gnn_tracking.analysis.graphs.OrphanCount attribute), [1] n_orphan_incorrect (gnn_tracking.analysis.graphs.OrphanCount attribute), [1] n_orphan_total (gnn_tracking.analysis.graphs.OrphanCount attribute), [1] n_particles (gnn_tracking.metrics.cluster_metrics.TrackingMetrics attribute) n_predicted_false() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) n_predicted_true() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) n_segments (gnn_tracking.analysis.graphs.TrackGraphInfo attribute), [1] n_true() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) O obj_from_or_to_hparams() (in module gnn_tracking.utils.lightning) object_loss() (gnn_tracking.metrics.losses.ObjectLoss method) ObjectLoss (class in gnn_tracking.metrics.losses) OCScanResults (class in gnn_tracking.postprocessing.dbscanscanner) on_test_epoch_end() (gnn_tracking.training.base.ImprovedLogLM method) on_train_batch_end() (gnn_tracking.utils.lightning.SimpleTqdmProgressBar method) on_train_epoch_end() (gnn_tracking.training.base.ImprovedLogLM method) on_train_epoch_start() (gnn_tracking.utils.lightning.SimpleTqdmProgressBar method) on_train_start() (gnn_tracking.training.callbacks.ExpandWandbConfig method) on_validation_end() (gnn_tracking.training.callbacks.PrintValidationMetrics method) on_validation_epoch_end() (gnn_tracking.training.base.ImprovedLogLM method) (gnn_tracking.training.ml.MLModule method) (gnn_tracking.utils.lightning.SimpleTqdmProgressBar method) OrphanCount (class in gnn_tracking.analysis.graphs) out_dim (gnn_tracking.models.graph_construction.MLGraphConstruction property) P perfect (gnn_tracking.metrics.cluster_metrics.TrackingMetrics attribute) PerfectECGraphTCN (class in gnn_tracking.models.track_condensation_networks) PerfectEdgeClassification (class in gnn_tracking.models.edge_classifier) pid (gnn_tracking.analysis.graphs.TrackGraphInfo attribute), [1] pivot_record_list() (in module gnn_tracking.utils.dictionaries) plot() (gnn_tracking.analysis.edge_classification.ThresholdTrackInfoPlot method) (gnn_tracking.graph_construction.k_scanner.KScanResults method) plot_100() (gnn_tracking.analysis.edge_classification.ThresholdTrackInfoPlot method) plot_2d() (gnn_tracking.utils.plotting.GraphPlotter method) plot_3d() (in module gnn_tracking.utils.plotting) plot_50() (gnn_tracking.analysis.edge_classification.ThresholdTrackInfoPlot method) plot_75() (gnn_tracking.analysis.edge_classification.ThresholdTrackInfoPlot method) plot_circles() (gnn_tracking.analysis.latent.SelectedPidsPlot static method) plot_collateral_ep() (gnn_tracking.analysis.latent.SelectedPidsPlot method) plot_collateral_latent() (gnn_tracking.analysis.latent.SelectedPidsPlot method) plot_ep_rv_uv() (gnn_tracking.utils.plotting.EventPlotter method) (gnn_tracking.utils.plotting.PointCloudPlotter method) plot_ep_rv_uv_all_sectors() (gnn_tracking.utils.plotting.PointCloudPlotter method) plot_ep_rv_uv_with_boundary() (gnn_tracking.utils.plotting.PointCloudPlotter method) plot_ep_rz_uv() (gnn_tracking.utils.plotting.GraphPlotter method) plot_errorline() (gnn_tracking.analysis.edge_classification.ThresholdTrackInfoPlot method) plot_hlines() (gnn_tracking.analysis.edge_classification.ThresholdTrackInfoPlot method) plot_line() (gnn_tracking.analysis.edge_classification.ThresholdTrackInfoPlot method) plot_mcc() (gnn_tracking.analysis.edge_classification.ThresholdTrackInfoPlot method) plot_other_hit_ep() (gnn_tracking.analysis.latent.SelectedPidsPlot method) plot_other_hit_latent() (gnn_tracking.analysis.latent.SelectedPidsPlot method) plot_rz() (gnn_tracking.utils.plotting.GraphPlotter method) (in module gnn_tracking.utils.plotting) plot_selected_pid_ep() (gnn_tracking.analysis.latent.SelectedPidsPlot method) plot_selected_pid_latent() (gnn_tracking.analysis.latent.SelectedPidsPlot method) plot_tpr_fpr() (gnn_tracking.analysis.edge_classification.ThresholdTrackInfoPlot method) PointCloudBuilder (class in gnn_tracking.preprocessing.point_cloud_builder) PointCloudPlotter (class in gnn_tracking.utils.plotting) PointCloudTCN (class in gnn_tracking.models.track_condensation_networks) PotentialLoss (class in gnn_tracking.metrics.losses) preprocess_detector() (in module gnn_tracking.preprocessing.exatrkx_cell_features) PreTrainedECGraphTCN (class in gnn_tracking.models.track_condensation_networks) PrintValidationMetrics (class in gnn_tracking.training.callbacks) process() (gnn_tracking.graph_construction.data_transformer.DataTransformer method) (gnn_tracking.graph_construction.graph_builder.GraphBuilder method) (gnn_tracking.preprocessing.point_cloud_builder.PointCloudBuilder method) process_directories() (gnn_tracking.graph_construction.data_transformer.DataTransformer method) R random_trial_name() (in module gnn_tracking.utils.nomenclature) remove_irrelevant_arguments() (in module gnn_tracking.utils.signature) reset() (gnn_tracking.graph_construction.k_scanner.GraphConstructionKNNScanner method) (gnn_tracking.postprocessing.clusterscanner.ClusterScanner method) (gnn_tracking.postprocessing.clusterscanner.CombinedClusterScanner method) (gnn_tracking.postprocessing.dbscanscanner.DBSCANHyperParamScanner method) reset_parameters() (gnn_tracking.models.dynamic_edge_conv.DynamicEdgeConv method) (gnn_tracking.models.edge_filter.EFMLP method) (gnn_tracking.models.graph_construction.GraphConstructionFCNN method) (gnn_tracking.models.mlp.MLP method) RESIDUAL_NETWORKS_BY_NAME (in module gnn_tracking.models.resin) ResidualNetwork (class in gnn_tracking.models.resin) ResIN (class in gnn_tracking.models.resin) restrict_to_subdetectors() (gnn_tracking.preprocessing.point_cloud_builder.PointCloudBuilder method) results (in module gnn_tracking.graph_construction.build_graphs_hpo) results_raw (gnn_tracking.graph_construction.k_scanner.GraphConstructionKNNScanner property) roc_auc_score() (in module gnn_tracking.metrics.binary_classification) S save_sub_hyperparameters() (in module gnn_tracking.utils.lightning) sector_hits() (gnn_tracking.preprocessing.point_cloud_builder.PointCloudBuilder method) select_edges() (gnn_tracking.graph_construction.graph_builder.GraphBuilder method) SelectedPidsPlot (class in gnn_tracking.analysis.latent) separate_init_kwargs() (in module gnn_tracking.utils.dictionaries) Sequential (class in gnn_tracking.models.meta) setup() (gnn_tracking.utils.loading.TestTrackingDataModule method) (gnn_tracking.utils.loading.TrackingDataModule method) setup_axes() (gnn_tracking.analysis.edge_classification.ThresholdTrackInfoPlot method) shortest_path_length_catch_no_path() (in module gnn_tracking.analysis.graphs) shortest_path_length_multi() (in module gnn_tracking.analysis.graphs) SimpleTqdmProgressBar (class in gnn_tracking.utils.lightning) Skip1ResidualNetwork (class in gnn_tracking.models.resin) Skip2ResidualNetwork (class in gnn_tracking.models.resin) SkipTopResidualNetwork (class in gnn_tracking.models.resin) sqconvex_combination() (in module gnn_tracking.models.resin) StandardError (class in gnn_tracking.utils.lightning) subdict_with_prefix_stripped() (in module gnn_tracking.utils.dictionaries) summarize_track_graph_info() (in module gnn_tracking.analysis.graphs) T TCModule (class in gnn_tracking.training.tc) test_dataloader() (gnn_tracking.utils.loading.TestTrackingDataModule method) (gnn_tracking.utils.loading.TrackingDataModule method) TestTrackingDataModule (class in gnn_tracking.utils.loading) theta_to_eta() (in module gnn_tracking.preprocessing.exatrkx_cell_features) ThresholdTrackInfoPlot (class in gnn_tracking.analysis.edge_classification) Timer (class in gnn_tracking.utils.timing) timing() (in module gnn_tracking.utils.timing) TN() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) TNR() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) to() (gnn_tracking.metrics.losses.LossFctType method) to_floats() (in module gnn_tracking.utils.dictionaries) to_pyg_data() (gnn_tracking.graph_construction.graph_builder.GraphBuilder method) (gnn_tracking.preprocessing.point_cloud_builder.PointCloudBuilder method) tolerate_additional_kwargs() (in module gnn_tracking.utils.signature) tolerate_some_oom_errors() (in module gnn_tracking.utils.oom) TP() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) TPR() (gnn_tracking.metrics.binary_classification.BinaryClassificationStats method) TrackGraphInfo (class in gnn_tracking.analysis.graphs) tracking_metric_df() (in module gnn_tracking.metrics.cluster_metrics) tracking_metrics() (in module gnn_tracking.metrics.cluster_metrics) tracking_metrics_data() (in module gnn_tracking.metrics.cluster_metrics) tracking_metrics_vs_eta() (in module gnn_tracking.metrics.cluster_metrics) tracking_metrics_vs_pt() (in module gnn_tracking.metrics.cluster_metrics) TrackingDataModule (class in gnn_tracking.utils.loading) TrackingDataset (class in gnn_tracking.utils.loading) TrackingMetrics (class in gnn_tracking.metrics.cluster_metrics) TrackingModule (class in gnn_tracking.training.base) train_dataloader() (gnn_tracking.utils.loading.TestTrackingDataModule method) (gnn_tracking.utils.loading.TrackingDataModule method) training_step() (gnn_tracking.training.ec.ECModule method) (gnn_tracking.training.ml.MLModule method) (gnn_tracking.training.tc.TCModule method) U unpack_loss_returns() (in module gnn_tracking.metrics.losses) update() (gnn_tracking.models.interaction_network.InteractionNetwork method) (gnn_tracking.utils.lightning.StandardError method) V val_dataloader() (gnn_tracking.utils.loading.TestTrackingDataModule method) (gnn_tracking.utils.loading.TrackingDataModule method) validation_step() (gnn_tracking.training.ec.ECModule method) (gnn_tracking.training.ml.MLModule method) (gnn_tracking.training.tc.TCModule method) Z zero_divide() (in module gnn_tracking.metrics.binary_classification) zero_division_gives_nan() (in module gnn_tracking.utils.math)
