Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | V | W | X | Z _ __call__() (gnn_tracking.graph_construction.k_scanner.GraphConstructionKNNScanner method) (gnn_tracking.metrics.cluster_metrics.ClusterMetricType 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) __getitem__() (gnn_tracking.utils.nomenclature.VariableManager method) __post_init__() (gnn_tracking.metrics.losses.MultiLossFctReturn method) __repr__() (gnn_tracking.models.dynamic_edge_conv.DynamicEdgeConv method) __str__() (gnn_tracking.utils.nomenclature.Variable method) __version__ (in module gnn_tracking) _alpha (gnn_tracking.metrics.losses.ec.HaughtyFocalLoss attribute) (gnn_tracking.models.mlp.ResFCNN attribute) (gnn_tracking.models.resin.ResidualNetwork attribute) _best_cluster_params (gnn_tracking.training.tc.TCModule attribute) _binary_focal_loss() (in module gnn_tracking.metrics.losses.ec) _build_graph() (gnn_tracking.metrics.losses.metric_learning.OldGraphConstructionHingeEmbeddingLoss method) _c_dfs (gnn_tracking.postprocessing.dbscanscanner.DBSCANPerformanceDetails attribute) _cluster_scan_input (gnn_tracking.training.tc.TCModule attribute) _collect_data (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) _collect_hidden_edge_embeds (gnn_tracking.models.resin.ResidualNetwork attribute) _color_mapper (gnn_tracking.analysis.latent.SelectedPidsPlot attribute) _configs (gnn_tracking.utils.loading.TrackingDataModule attribute) _cpus (gnn_tracking.utils.loading.TrackingDataModule attribute) _cut_to_pixel_cms() (gnn_tracking.utils.loading.TrackingDataset method) _data_list (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) _datasets (gnn_tracking.utils.loading.TrackingDataModule attribute) _decoder (gnn_tracking.models.graph_construction.GraphConstructionResIN attribute) (gnn_tracking.models.mlp.ResFCNN attribute) _DEFAULT_FEATURE_SCALE (in module gnn_tracking.preprocessing.point_cloud_builder) _DEFAULT_KS (in module gnn_tracking.graph_construction.k_scanner) _detector (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) _df (gnn_tracking.postprocessing.dbscanscanner.OCScanResults attribute) _df_mean (gnn_tracking.postprocessing.dbscanscanner.OCScanResults attribute) _distances (gnn_tracking.postprocessing.fastrescanner.DBSCANFastRescan attribute) _e_tilde (gnn_tracking.models.interaction_network.InteractionNetwork attribute) _ec_hit_mask (gnn_tracking.analysis.latent.SelectedPidsPlot attribute) _edge_augmentation (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) _edge_batch_norms (gnn_tracking.models.resin.Skip2ResidualNetwork attribute) _edge_encoder (gnn_tracking.models.graph_construction.GraphConstructionResIN attribute) _edges (gnn_tracking.postprocessing.fastrescanner.DBSCANFastRescan attribute) _ef (gnn_tracking.models.graph_construction.MLGraphConstruction attribute) _encoder (gnn_tracking.models.mlp.ResFCNN attribute) _eta (gnn_tracking.analysis.latent.SelectedPidsPlot attribute) _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) _evaluate_tracking_metrics_upper_bounds() (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 property) _find_loggers() (gnn_tracking.training.callbacks.ExpandWandbConfig method) _first_occurrences() (in module gnn_tracking.metrics.losses.oc) _fix_datatypes() (gnn_tracking.utils.loading.TrackingDataModule static method) _forward() (gnn_tracking.metrics.losses.ec.EdgeWeightBCELoss static method) (gnn_tracking.metrics.losses.ec.EdgeWeightFocalLoss method) (gnn_tracking.metrics.losses.ec.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) _gamma (gnn_tracking.metrics.losses.ec.HaughtyFocalLoss attribute) _get_alphas_first_occurences() (in module gnn_tracking.metrics.losses.oc) _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.BasePointCloudBuilder method) _get_edges() (gnn_tracking.metrics.losses.metric_learning.GraphConstructionHingeEmbeddingLoss method) _get_foms_at_target() (gnn_tracking.graph_construction.k_scanner.KScanResults method) _get_paths() (gnn_tracking.utils.loading.TrackingDataset method) _get_target_k() (gnn_tracking.graph_construction.k_scanner.KScanResults method) _get_va() (in module gnn_tracking.metrics.losses.oc) _get_vr_rg() (in module gnn_tracking.metrics.losses.oc) _gtcn (gnn_tracking.models.track_condensation_networks.GraphTCN attribute) (gnn_tracking.models.track_condensation_networks.GraphTCNForMLGCPipeline attribute) (gnn_tracking.models.track_condensation_networks.PerfectECGraphTCN attribute) (gnn_tracking.models.track_condensation_networks.PreTrainedECGraphTCN attribute) _h_dfs (gnn_tracking.postprocessing.dbscanscanner.DBSCANPerformanceDetails attribute) _hinge_loss_components() (in module gnn_tracking.metrics.losses.metric_learning) _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) _labels (gnn_tracking.analysis.latent.SelectedPidsPlot attribute) _latent_normalization (gnn_tracking.models.graph_construction.GraphConstructionFCNN attribute) (gnn_tracking.models.graph_construction.GraphConstructionHeteroEncResFCNN attribute) (gnn_tracking.models.graph_construction.GraphConstructionHeteroResFCNN attribute) (gnn_tracking.models.graph_construction.GraphConstructionResFCNN attribute) (gnn_tracking.models.graph_construction.GraphConstructionResIN attribute) (gnn_tracking.models.track_condensation_networks.ModularGraphTCN attribute) _latex (gnn_tracking.utils.nomenclature.Variable attribute) _layers (gnn_tracking.models.meta.Sequential attribute) (gnn_tracking.models.mlp.ResFCNN attribute) _legend_items (gnn_tracking.analysis.efficiencies.PerformanceComparisonPlot attribute) (gnn_tracking.analysis.efficiencies.PerformancePlot attribute) _log_errors() (gnn_tracking.training.base.ImprovedLogLM method) _loss (gnn_tracking.metrics.losses.LossClones attribute) _make_pt_cut() (gnn_tracking.utils.loading.TrackingDataset method) _max_eps (gnn_tracking.postprocessing.fastrescanner.DBSCANFastRescan attribute) _ml (gnn_tracking.models.graph_construction.MLGraphConstruction attribute) (gnn_tracking.models.graph_construction.MLPCTransformer attribute) _model (gnn_tracking.graph_construction.data_transformer.ECCut attribute) _mse() (gnn_tracking.metrics.losses.oc.ObjectLoss static method) _n_jobs (gnn_tracking.postprocessing.fastrescanner.DBSCANFastRescan attribute) _node_batch_norms (gnn_tracking.models.resin.Skip2ResidualNetwork attribute) _node_encoder (gnn_tracking.models.graph_construction.GraphConstructionResIN attribute) _old_hinge_loss_components() (in module gnn_tracking.metrics.losses.metric_learning) _output (gnn_tracking.metrics.binary_classification.BinaryClassificationStats attribute) _P (in module gnn_tracking.utils.dictionaries) _parameters (gnn_tracking.postprocessing.dbscanscanner.OCScanResults attribute) _phi (gnn_tracking.analysis.latent.SelectedPidsPlot attribute) _pids (gnn_tracking.analysis.latent.SelectedPidsPlot attribute) _predicted_false (gnn_tracking.metrics.binary_classification.BinaryClassificationStats property) _predicted_true (gnn_tracking.metrics.binary_classification.BinaryClassificationStats property) _prefixes (gnn_tracking.metrics.losses.LossClones attribute) _process_sectors() (gnn_tracking.preprocessing.point_cloud_builder.TrackMLPointCloudBuilder method) _processed_paths (gnn_tracking.utils.loading.TrackingDataset attribute) _pt_thld (gnn_tracking.metrics.losses.ec.HaughtyFocalLoss attribute) _radius_graph_condensation_loss() (in module gnn_tracking.metrics.losses.oc) _remove_intersecting (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) _reset_graph() (gnn_tracking.postprocessing.fastrescanner.DBSCANFastRescan method) _reset_layer_parameters() (gnn_tracking.models.edge_filter.EFMLP static method) (gnn_tracking.models.mlp.ResFCNN static method) _reset_trials() (gnn_tracking.postprocessing.dbscanscanner.DBSCANHyperParamScanner method) (gnn_tracking.postprocessing.dbscanscanner.DBSCANHyperParamScannerFixed method) _residual_layer (gnn_tracking.models.resin.SkipTopResidualNetwork attribute) _residual_type (gnn_tracking.models.resin.ResIN attribute) _resin (gnn_tracking.models.graph_construction.GraphConstructionResIN attribute) _results (gnn_tracking.graph_construction.k_scanner.GraphConstructionKNNScanner attribute) (gnn_tracking.postprocessing.dbscanscanner.DBSCANHyperParamScanner attribute) _rng (gnn_tracking.postprocessing.dbscanscanner.DBSCANHyperParamScanner attribute) _save_hparams() (gnn_tracking.graph_construction.data_transformer.DataTransformer method) _scanners (gnn_tracking.postprocessing.clusterscanner.CombinedClusterScanner attribute) _selected_pid_mask (gnn_tracking.analysis.latent.SelectedPidsPlot attribute) _selected_pids (gnn_tracking.analysis.latent.SelectedPidsPlot attribute) _sklearn_signature_wrap() (in module gnn_tracking.metrics.cluster_metrics) _spline (gnn_tracking.graph_construction.k_scanner.KScanResults property) _sqconvex_combination() (in module gnn_tracking.models.resin) _square_distances() (in module gnn_tracking.metrics.losses.oc) _tensorboard_logger (gnn_tracking.training.callbacks.ExpandWandbConfig attribute) _thld (gnn_tracking.metrics.binary_classification.BinaryClassificationStats attribute) _tracking_metrics_nan_results (in module gnn_tracking.metrics.cluster_metrics) _transform (gnn_tracking.graph_construction.data_transformer.DataTransformer attribute) _trials (gnn_tracking.postprocessing.dbscanscanner.DBSCANHyperParamScanner attribute) (gnn_tracking.postprocessing.dbscanscanner.DBSCANHyperParamScannerFixed attribute) _true (gnn_tracking.metrics.binary_classification.BinaryClassificationStats property) _uncertainties (gnn_tracking.training.base.ImprovedLogLM attribute) _validate_config() (gnn_tracking.models.graph_construction.MLGraphConstruction method) _variables (gnn_tracking.utils.nomenclature.VariableManager attribute) _wandb_logger (gnn_tracking.training.callbacks.ExpandWandbConfig attribute) _x (gnn_tracking.analysis.latent.SelectedPidsPlot attribute) _y (gnn_tracking.metrics.binary_classification.BinaryClassificationStats attribute) A acc (gnn_tracking.metrics.binary_classification.BinaryClassificationStats property) act (gnn_tracking.models.mlp.ResMLP attribute) add() (gnn_tracking.utils.nomenclature.VariableManager method) add_blocked() (gnn_tracking.analysis.efficiencies.PerformanceComparisonPlot method) (gnn_tracking.analysis.efficiencies.PerformancePlot 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) (gnn_tracking.analysis.efficiencies.PerformanceComparisonPlot method) (gnn_tracking.analysis.efficiencies.PerformancePlot method) (gnn_tracking.analysis.plotutils.Plot method) add_true_edges (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) add_watermark() (in module gnn_tracking.analysis.plotutils) aggregator (gnn_tracking.models.edge_filter.EFDeepSet attribute) append_cell_features() (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder method) append_coordinates() (gnn_tracking.utils.plotting.EventPlotter method) append_features() (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder method) append_n_layers_hit() (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder static 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) assign_background_track_ids() (gnn_tracking.preprocessing.point_cloud_builder.CMSPointCloudBuilder static method) augment_hit_features() (in module gnn_tracking.preprocessing.exatrkx_cell_features) ax (gnn_tracking.analysis.edge_classification.ThresholdTrackInfoPlot attribute) (gnn_tracking.analysis.plotutils.Plot attribute) B B (gnn_tracking.models.track_condensation_networks.PointCloudTCN attribute) backward() (gnn_tracking.training.base.TrackingModule method) balanced_acc (gnn_tracking.metrics.binary_classification.BinaryClassificationStats property) bar (gnn_tracking.utils.lightning.SimpleTqdmProgressBar attribute) BasePointCloudBuilder (class in gnn_tracking.preprocessing.point_cloud_builder) beta (gnn_tracking.models.mlp.ResMLP attribute) binary_focal_loss() (in module gnn_tracking.metrics.losses.ec) BinaryClassificationStats (class in gnn_tracking.metrics.binary_classification) build_edges() (gnn_tracking.graph_construction.graph_builder.GraphBuilder method) builder_params (gnn_tracking.utils.loading.TrackingDataModule attribute) 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.BasePointCloudBuilder static 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) cluster_scanner (gnn_tracking.training.tc.TCModule attribute) ClusterMetricType (class in gnn_tracking.metrics.cluster_metrics) ClusterScanner (class in gnn_tracking.postprocessing.clusterscanner) CMSPointCloudBuilder (class in gnn_tracking.preprocessing.point_cloud_builder) collect_all_ec_stats() (in module gnn_tracking.analysis.edge_classification) collect_measurements() (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder method) colors (gnn_tracking.utils.plotting.PointCloudPlotter attribute) 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) condensation_loss_tiger() (in module gnn_tracking.metrics.losses.oc) CondensationLossRG (class in gnn_tracking.metrics.losses.oc) CondensationLossTiger (class in gnn_tracking.metrics.losses.oc) 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) (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) data_preproc() (gnn_tracking.training.base.TrackingModule method) data_type (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) DatasetReader (class in gnn_tracking.preprocessing.point_cloud_builder) datasets (gnn_tracking.utils.loading.TestTrackingDataModule attribute) (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) decoder (gnn_tracking.models.edge_filter.EFMLP attribute) DEFAULT_FEATURES (in module gnn_tracking.preprocessing.point_cloud_builder) (in module gnn_tracking.utils.loading) denote_pt() (in module gnn_tracking.utils.nomenclature) depth (gnn_tracking.models.mlp.ResMLP attribute) detector (gnn_tracking.preprocessing.exatrkx_cell_features.DetectorPixelSize attribute) (gnn_tracking.preprocessing.exatrkx_cell_features.DetectorRotations attribute) (gnn_tracking.preprocessing.exatrkx_cell_features.DetectorThicknesses attribute) 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.analysis.edge_classification.ThresholdTrackInfoPlot attribute) (gnn_tracking.analysis.efficiencies.PerformancePlot attribute) (gnn_tracking.analysis.efficiencies.TracksVsDBSCANPlot attribute) (gnn_tracking.graph_construction.k_scanner.KScanResults attribute) (gnn_tracking.postprocessing.dbscanscanner.OCScanResults property) df_mean (gnn_tracking.postprocessing.dbscanscanner.OCScanResults property) df_ul (gnn_tracking.analysis.efficiencies.PerformancePlot attribute) directed (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) 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) dR_max (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) DummyMultiLoss (class in gnn_tracking.metrics.losses) DynamicEdgeConv (class in gnn_tracking.models.dynamic_edge_conv) E ec (gnn_tracking.models.track_condensation_networks.ModularGraphTCN attribute) ec_edge_encoder (gnn_tracking.models.edge_classifier.ECForGraphTCN attribute) ec_node_encoder (gnn_tracking.models.edge_classifier.ECForGraphTCN attribute) ec_resin (gnn_tracking.models.edge_classifier.ECForGraphTCN attribute) 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) edge_conv (gnn_tracking.models.track_condensation_networks.INConvBlock attribute) edge_dim (gnn_tracking.models.resin.ResIN attribute) edge_encoder (gnn_tracking.models.track_condensation_networks.INConvBlock attribute) edge_index (gnn_tracking.models.dynamic_edge_conv.DynamicEdgeConv attribute) EdgeWeightBCELoss (class in gnn_tracking.metrics.losses.ec) EdgeWeightFocalLoss (class in gnn_tracking.metrics.losses.ec) EFDeepSet (class in gnn_tracking.models.edge_filter) EFMLP (class in gnn_tracking.models.edge_filter) enabled (gnn_tracking.utils.lightning.SimpleTqdmProgressBar attribute) encoder (gnn_tracking.models.edge_filter.EFMLP attribute) (gnn_tracking.models.graph_construction.GraphConstructionHeteroEncResFCNN attribute) eta_0 (gnn_tracking.models.mlp.ResMLP attribute) EventPlotter (class in gnn_tracking.utils.plotting) exists (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) expand_grid() (in module gnn_tracking.utils.dictionaries) ExpandWandbConfig (class in gnn_tracking.training.callbacks) extra_metrics (gnn_tracking.metrics.losses.MultiLossFctReturn attribute) extract_dir_new() (in module gnn_tracking.preprocessing.exatrkx_cell_features) F F1 (gnn_tracking.metrics.binary_classification.BinaryClassificationStats property) 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) false_below_pt (gnn_tracking.models.edge_classifier.PerfectEdgeClassification attribute) falsify_low_pt_edges() (in module gnn_tracking.metrics.losses.ec) FalsifyLowPtEdgeWeightLoss (class in gnn_tracking.metrics.losses.ec) fcnn (gnn_tracking.models.graph_construction.GraphConstructionHeteroEncResFCNN attribute) feature_names (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) (gnn_tracking.preprocessing.point_cloud_builder.MDPointCloudBuilder attribute) feature_scale (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) (gnn_tracking.preprocessing.point_cloud_builder.MDPointCloudBuilder attribute) feature_subset_names (gnn_tracking.utils.loading.TrackingDataModule attribute) file_number (gnn_tracking.utils.loading.TrackingDataset attribute) FILENAME (gnn_tracking.analysis.plotutils.Plot attribute) 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 property) FNR (gnn_tracking.metrics.binary_classification.BinaryClassificationStats property) 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.DummyMultiLoss method) (gnn_tracking.metrics.losses.ec.FalsifyLowPtEdgeWeightLoss method) (gnn_tracking.metrics.losses.ec.HaughtyFocalLoss method) (gnn_tracking.metrics.losses.LossClones method) (gnn_tracking.metrics.losses.metric_learning.GraphConstructionHingeEmbeddingLoss method) (gnn_tracking.metrics.losses.metric_learning.OldGraphConstructionHingeEmbeddingLoss method) (gnn_tracking.metrics.losses.MultiLossFct method) (gnn_tracking.metrics.losses.oc.CondensationLossRG method) (gnn_tracking.metrics.losses.oc.CondensationLossTiger method) (gnn_tracking.metrics.losses.oc.ObjectLoss 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.GraphConstructionHeteroEncResFCNN method) (gnn_tracking.models.graph_construction.GraphConstructionHeteroResFCNN method) (gnn_tracking.models.graph_construction.GraphConstructionResFCNN 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.HeterogeneousResFCNN method) (gnn_tracking.models.mlp.MLP method) (gnn_tracking.models.mlp.ResFCNN method) (gnn_tracking.models.mlp.ResMLP method) (gnn_tracking.models.noise_classification.TruthNoiseClassifierModel method) (gnn_tracking.models.noise_classification.WithNoiseClassification 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.GraphTCNForMLGCPipeline 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 property) FPR (gnn_tracking.metrics.binary_classification.BinaryClassificationStats property) 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 gamma (gnn_tracking.models.mlp.ResMLP attribute) gamma_0 (gnn_tracking.models.mlp.ResMLP attribute) gc_scanner (gnn_tracking.training.ml.MLModule attribute) 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_good_node_mask_tensors() (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_lr() (gnn_tracking.models.mlp.ResMLP 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.TrackMLPointCloudBuilder 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_mask() (in module gnn_tracking.models.mlp) 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() (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder static method) 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.efficiencies module gnn_tracking.analysis.graphs module gnn_tracking.analysis.latent module gnn_tracking.analysis.plotutils 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.metrics.losses.ec module gnn_tracking.metrics.losses.metric_learning module gnn_tracking.metrics.losses.oc 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.noise_classification 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) GraphConstructionHeteroEncResFCNN (class in gnn_tracking.models.graph_construction) GraphConstructionHeteroResFCNN (class in gnn_tracking.models.graph_construction) GraphConstructionHingeEmbeddingLoss (class in gnn_tracking.metrics.losses.metric_learning) GraphConstructionKNNScanner (class in gnn_tracking.graph_construction.k_scanner) GraphConstructionResFCNN (class in gnn_tracking.models.graph_construction) GraphConstructionResIN (class in gnn_tracking.models.graph_construction) GraphPlotter (class in gnn_tracking.utils.plotting) graphs (gnn_tracking.utils.loading.TestTrackingDataModule attribute) GraphTCN (class in gnn_tracking.models.track_condensation_networks) GraphTCNForMLGCPipeline (class in gnn_tracking.models.track_condensation_networks) guess_device() (in module gnn_tracking.utils.device) H HaughtyFocalLoss (class in gnn_tracking.metrics.losses.ec) hc_edge_encoder (gnn_tracking.models.track_condensation_networks.ModularGraphTCN attribute) hc_in (gnn_tracking.models.track_condensation_networks.ModularGraphTCN attribute) HeterogeneousResFCNN (class in gnn_tracking.models.mlp) 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) in_dim (gnn_tracking.models.mlp.ResMLP attribute) INConvBlock (class in gnn_tracking.models.track_condensation_networks) indir (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) (gnn_tracking.utils.plotting.EventPlotter attribute) (gnn_tracking.utils.plotting.GraphPlotter attribute) (gnn_tracking.utils.plotting.PointCloudPlotter attribute) infiles (gnn_tracking.utils.plotting.EventPlotter attribute) (gnn_tracking.utils.plotting.PointCloudPlotter attribute) initial_outdir (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) input_tree (gnn_tracking.preprocessing.point_cloud_builder.MDPointCloudBuilder attribute) 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 k (gnn_tracking.models.dynamic_edge_conv.DynamicEdgeConv attribute) 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 latent_dim (gnn_tracking.models.edge_classifier.ECForGraphTCN attribute) latex (gnn_tracking.utils.nomenclature.Variable property) layers (gnn_tracking.models.edge_filter.EFMLP attribute) (gnn_tracking.models.mlp.MLP attribute) (gnn_tracking.models.mlp.ResMLP attribute) (gnn_tracking.models.resin.ResidualNetwork attribute) (gnn_tracking.models.track_condensation_networks.INConvBlock attribute) (gnn_tracking.models.track_condensation_networks.PointCloudTCN attribute) 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_new_cms_mc_file() (gnn_tracking.preprocessing.point_cloud_builder.CMSPointCloudBuilder method) load_obj_from_hparams() (in module gnn_tracking.utils.lightning) load_trackml_event() (gnn_tracking.preprocessing.point_cloud_builder.TrackMLPointCloudBuilder static method) LOG_DEFAULT_LEVEL (in module gnn_tracking.utils.log) log_dict_with_errors() (gnn_tracking.training.base.ImprovedLogLM method) logg (gnn_tracking.training.base.TrackingModule attribute) logger (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) (in module gnn_tracking.metrics.losses) (in module gnn_tracking.utils.log) loss (gnn_tracking.metrics.losses.MultiLossFctReturn property) loss_dct (gnn_tracking.metrics.losses.MultiLossFctReturn attribute) loss_fct (gnn_tracking.training.ec.ECModule attribute) (gnn_tracking.training.ml.MLModule attribute) (gnn_tracking.training.tc.TCModule attribute) LossClones (class in gnn_tracking.metrics.losses) lr (gnn_tracking.models.mlp.ResMLP attribute) LS_COLS (in module gnn_tracking.preprocessing.point_cloud_builder) M MCC (gnn_tracking.metrics.binary_classification.BinaryClassificationStats property) MD_COLS (in module gnn_tracking.preprocessing.point_cloud_builder) MD_FEATURES (in module gnn_tracking.preprocessing.point_cloud_builder) MDPointCloudBuilder (class in gnn_tracking.preprocessing.point_cloud_builder) measurement_mode (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) MEASUREMENT_OUTDIR (in module gnn_tracking.graph_construction.build_graphs_hpo) measurements (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) 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) mode (gnn_tracking.metrics.losses.oc.ObjectLoss attribute) model (gnn_tracking.models.noise_classification.WithNoiseClassification attribute) (gnn_tracking.training.base.TrackingModule attribute) ModularGraphTCN (class in gnn_tracking.models.track_condensation_networks) module gnn_tracking gnn_tracking.analysis gnn_tracking.analysis.edge_classification gnn_tracking.analysis.efficiencies gnn_tracking.analysis.graphs gnn_tracking.analysis.latent gnn_tracking.analysis.plotutils 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.metrics.losses.ec gnn_tracking.metrics.losses.metric_learning gnn_tracking.metrics.losses.oc 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.noise_classification 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 MultiLossFct (class in gnn_tracking.metrics.losses) MultiLossFctReturn (class in gnn_tracking.metrics.losses) 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 property) 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 property) n_predicted_true (gnn_tracking.metrics.binary_classification.BinaryClassificationStats property) n_sectors (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) (gnn_tracking.utils.plotting.GraphPlotter attribute) (gnn_tracking.utils.plotting.PointCloudPlotter attribute) n_segments (gnn_tracking.analysis.graphs.TrackGraphInfo attribute), [1] n_true (gnn_tracking.metrics.binary_classification.BinaryClassificationStats property) name (gnn_tracking.utils.nomenclature.Variable attribute) network (gnn_tracking.models.resin.ResIN attribute) nn (gnn_tracking.models.dynamic_edge_conv.DynamicEdgeConv attribute) node_dim (gnn_tracking.models.resin.ResIN attribute) node_encoder (gnn_tracking.models.edge_filter.EFDeepSet attribute) (gnn_tracking.models.track_condensation_networks.INConvBlock attribute) noise_model (gnn_tracking.models.noise_classification.WithNoiseClassification attribute) num_workers (gnn_tracking.models.dynamic_edge_conv.DynamicEdgeConv attribute) O obj_from_or_to_hparams() (in module gnn_tracking.utils.lightning) object_loss() (gnn_tracking.metrics.losses.oc.ObjectLoss method) object_model (gnn_tracking.models.interaction_network.InteractionNetwork attribute) ObjectLoss (class in gnn_tracking.metrics.losses.oc) OCScanResults (class in gnn_tracking.postprocessing.dbscanscanner) OldGraphConstructionHingeEmbeddingLoss (class in gnn_tracking.metrics.losses.metric_learning) 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) optimizer (gnn_tracking.training.base.TrackingModule attribute) OrphanCount (class in gnn_tracking.analysis.graphs) out_dim (gnn_tracking.models.graph_construction.MLGraphConstruction property) (gnn_tracking.models.mlp.ResMLP attribute) outdir (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) outfiles (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) P p_beta (gnn_tracking.models.track_condensation_networks.ModularGraphTCN attribute) p_cluster (gnn_tracking.models.track_condensation_networks.ModularGraphTCN attribute) pc_builder_dict (in module gnn_tracking.preprocessing.build_point_clouds) 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) PerformanceComparisonPlot (class in gnn_tracking.analysis.efficiencies) PerformancePlot (class in gnn_tracking.analysis.efficiencies) phi_slope_max (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) pid (gnn_tracking.analysis.graphs.TrackGraphInfo attribute), [1] pivot_record_list() (in module gnn_tracking.utils.dictionaries) pixel_fcnn (gnn_tracking.models.mlp.HeterogeneousResFCNN attribute) pixel_only (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) Plot (class in gnn_tracking.analysis.plotutils) 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) plot_var() (gnn_tracking.analysis.efficiencies.PerformanceComparisonPlot method) (gnn_tracking.analysis.efficiencies.PerformancePlot method) (gnn_tracking.analysis.efficiencies.TracksVsDBSCANPlot method) point_cloud_builder (gnn_tracking.utils.loading.TrackingDataset attribute) PointCloudPlotter (class in gnn_tracking.utils.plotting) PointCloudTCN (class in gnn_tracking.models.track_condensation_networks) predict_dataloader() (gnn_tracking.utils.loading.TrackingDataModule method) prefixes (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) preproc (gnn_tracking.training.base.TrackingModule attribute) preprocess_detector() (in module gnn_tracking.preprocessing.exatrkx_cell_features) PreTrainedECGraphTCN (class in gnn_tracking.models.track_condensation_networks) prev_file_number (gnn_tracking.utils.loading.TrackingDataset attribute) print_validation_results (gnn_tracking.training.base.ImprovedLogLM attribute) 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.BasePointCloudBuilder method) process_directories() (gnn_tracking.graph_construction.data_transformer.DataTransformer method) process_event() (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder method) (gnn_tracking.preprocessing.point_cloud_builder.CMSPointCloudBuilder method) (gnn_tracking.preprocessing.point_cloud_builder.MDPointCloudBuilder method) (gnn_tracking.preprocessing.point_cloud_builder.TrackMLPointCloudBuilder method) pt_cut (gnn_tracking.utils.loading.TrackingDataModule attribute) (gnn_tracking.utils.loading.TrackingDataset attribute) R random_trial_name() (in module gnn_tracking.utils.nomenclature) read_event() (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder method) (gnn_tracking.preprocessing.point_cloud_builder.CMSPointCloudBuilder method) (gnn_tracking.preprocessing.point_cloud_builder.DatasetReader method) (gnn_tracking.preprocessing.point_cloud_builder.MDPointCloudBuilder method) (gnn_tracking.preprocessing.point_cloud_builder.TrackMLPointCloudBuilder method) redo (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) relational_model (gnn_tracking.models.interaction_network.InteractionNetwork attribute) relu (gnn_tracking.models.edge_classifier.ECForGraphTCN attribute) (gnn_tracking.models.track_condensation_networks.INConvBlock attribute) (gnn_tracking.models.track_condensation_networks.ModularGraphTCN attribute) remove_irrelevant_arguments() (in module gnn_tracking.utils.signature) remove_noise (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) 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.mlp.MLP method) (gnn_tracking.models.mlp.ResMLP method) ResFCNN (class in gnn_tracking.models.mlp) RESIDUAL_NETWORKS_BY_NAME (in module gnn_tracking.models.resin) ResidualNetwork (class in gnn_tracking.models.resin) ResIN (class in gnn_tracking.models.resin) ResMLP (class in gnn_tracking.models.mlp) restrict_to_subdetectors() (gnn_tracking.preprocessing.point_cloud_builder.TrackMLPointCloudBuilder method) results (in module gnn_tracking.graph_construction.build_graphs_hpo) results_raw (gnn_tracking.graph_construction.k_scanner.GraphConstructionKNNScanner property) return_data (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) roc_auc_score() (in module gnn_tracking.metrics.binary_classification) S save() (gnn_tracking.analysis.plotutils.Plot method) save_output_file() (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder method) save_sub_hyperparameters() (in module gnn_tracking.utils.lightning) scheduler (gnn_tracking.training.base.TrackingModule attribute) sector_di (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) sector_ds (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) sector_hits() (gnn_tracking.preprocessing.point_cloud_builder.TrackMLPointCloudBuilder method) sector_results (gnn_tracking.utils.loading.TrackingDataset attribute) 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) stats (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) strip_fcnn (gnn_tracking.models.mlp.HeterogeneousResFCNN attribute) subdict_with_prefix_stripped() (in module gnn_tracking.utils.dictionaries) summarize_track_graph_info() (in module gnn_tracking.analysis.graphs) T t0 (gnn_tracking.utils.timing.Timer attribute) targets (gnn_tracking.graph_construction.k_scanner.KScanResults attribute) 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) thld (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) 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 property) TNR (gnn_tracking.metrics.binary_classification.BinaryClassificationStats property) tnr (gnn_tracking.models.edge_classifier.PerfectEdgeClassification attribute) 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.BasePointCloudBuilder 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 property) TPR (gnn_tracking.metrics.binary_classification.BinaryClassificationStats property) tpr (gnn_tracking.models.edge_classifier.PerfectEdgeClassification attribute) 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) TrackMLPointCloudBuilder (class in gnn_tracking.preprocessing.point_cloud_builder) TracksVsDBSCANPlot (class in gnn_tracking.analysis.efficiencies) 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) TruthNoiseClassifierModel (class in gnn_tracking.models.noise_classification) U update() (gnn_tracking.models.interaction_network.InteractionNetwork method) (gnn_tracking.utils.lightning.StandardError method) use_acat_standard (gnn_tracking.models.track_condensation_networks.ModularGraphTCN attribute) 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) var (gnn_tracking.analysis.efficiencies.PerformanceComparisonPlot attribute) Variable (class in gnn_tracking.utils.nomenclature) variable_manager (in module gnn_tracking.utils.nomenclature) VariableManager (class in gnn_tracking.utils.nomenclature) W W (gnn_tracking.models.edge_classifier.ECForGraphTCN attribute) weight_dct (gnn_tracking.metrics.losses.MultiLossFctReturn attribute) weighted_losses (gnn_tracking.metrics.losses.MultiLossFctReturn property) width (gnn_tracking.models.mlp.ResMLP attribute) WithNoiseClassification (class in gnn_tracking.models.noise_classification) write_output (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) (gnn_tracking.preprocessing.point_cloud_builder.BasePointCloudBuilder attribute) X X (gnn_tracking.models.track_condensation_networks.PointCloudTCN attribute) x (gnn_tracking.postprocessing.fastrescanner.DBSCANFastRescan attribute) xs (gnn_tracking.analysis.efficiencies.PerformanceComparisonPlot attribute) (gnn_tracking.analysis.efficiencies.PerformancePlot attribute) Z z0_max (gnn_tracking.graph_construction.graph_builder.GraphBuilder attribute) zero_divide() (in module gnn_tracking.metrics.binary_classification) zero_division_gives_nan() (in module gnn_tracking.utils.math)