Index

A

AbsConv1dHead (class in lumin.nn.models.blocks.head) AbsCyclicCallback (class in lumin.nn.callbacks.cyclic_callbacks) AbsEndcap (class in lumin.nn.models.blocks.endcap) AdaptiveAvgMaxConcatPool1d (class in lumin.nn.models.blocks.conv_blocks) AdaptiveAvgMaxConcatPool2d (class in lumin.nn.models.blocks.conv_blocks) AdaptiveAvgMaxConcatPool3d (class in lumin.nn.models.blocks.conv_blocks) add_abs_mom() (in module lumin.data_processing.hep_proc) add_energy() (in module lumin.data_processing.hep_proc) add_ignore() (lumin.nn.data.fold_yielder.FoldYielder method) (lumin.nn.data.fold_yielder.TorchGeometricFoldYielder method) add_input_pipe() (lumin.nn.data.fold_yielder.FoldYielder method) (lumin.nn.ensemble.ensemble.Ensemble method) add_input_pipe_from_file() (lumin.nn.data.fold_yielder.FoldYielder method)

add_mass() (in module lumin.data_processing.hep_proc) add_matrix_pipe() (lumin.nn.data.fold_yielder.FoldYielder method) add_matrix_pipe_from_file() (lumin.nn.data.fold_yielder.FoldYielder method) add_meta_data() (in module lumin.data_processing.file_proc) add_mt() (in module lumin.data_processing.hep_proc) add_output_pipe() (lumin.nn.data.fold_yielder.FoldYielder method) (lumin.nn.ensemble.ensemble.Ensemble method) add_output_pipe_from_file() (lumin.nn.data.fold_yielder.FoldYielder method) AMS (class in lumin.nn.metrics.class_eval) ams_scan_quick() (in module lumin.evaluation.ams) ams_scan_slow() (in module lumin.evaluation.ams) auto_filter_on_linear_correlation() (in module lumin.optimisation.features) auto_filter_on_mutual_dependence() (in module lumin.optimisation.features) AutoExtractLorentzBoostNet (class in lumin.nn.models.blocks.head)

B

BackwardHook (class in lumin.utils.misc) BatchYielder (class in lumin.nn.data.batch_yielder) bin_binary_class_pred() (in module lumin.inference.summary_stat) binary_class_cut_by_ams() (in module lumin.optimisation.threshold) BinaryAccuracy (class in lumin.nn.metrics.class_eval)

BinaryLabelSmooth (class in lumin.nn.callbacks.data_callbacks) boost() (in module lumin.data_processing.hep_proc) boost2cm() (in module lumin.data_processing.hep_proc) bootstrap_stats() (in module lumin.utils.statistics) BootstrapResample (class in lumin.nn.callbacks.data_callbacks) build_model() (lumin.nn.models.model_builder.ModelBuilder method)

C

calc_ams() (in module lumin.evaluation.ams) calc_ams_torch() (in module lumin.evaluation.ams) calc_emb_szs() (lumin.nn.models.helpers.CatEmbedder method) calc_pair_mass() (in module lumin.data_processing.hep_proc) Callback (class in lumin.nn.callbacks.callback) CatEmbedder (class in lumin.nn.models.helpers) CatEmbHead (class in lumin.nn.models.blocks.head) check_out_sz() (lumin.nn.models.blocks.head.AbsConv1dHead method) (lumin.nn.models.blocks.head.LorentzBoostNet method) check_val_set() (in module lumin.utils.data)

ClassRegMulti (class in lumin.nn.models.blocks.tail) close() (lumin.nn.data.fold_yielder.FoldYielder method) (lumin.nn.data.fold_yielder.TorchGeometricFoldYielder method) columns() (lumin.nn.data.fold_yielder.FoldYielder method) (lumin.nn.data.fold_yielder.TorchGeometricFoldYielder method) compare_events() (in module lumin.plotting.data_viewing) Conv1DBlock (class in lumin.nn.models.blocks.conv_blocks) cos_delta() (in module lumin.data_processing.hep_proc) CycleLR (class in lumin.nn.callbacks.cyclic_callbacks) CycleMom (class in lumin.nn.callbacks.cyclic_callbacks) CycleStep (class in lumin.nn.callbacks.cyclic_callbacks)

D

delta_phi() (in module lumin.data_processing.hep_proc) delta_r() (in module lumin.data_processing.hep_proc)

delta_r_boosted() (in module lumin.data_processing.hep_proc) df2foldfile() (in module lumin.data_processing.file_proc)

E

EarlyStopping (class in lumin.nn.callbacks.monitors) Ensemble (class in lumin.nn.ensemble.ensemble) EpochSaver (class in lumin.nn.callbacks.monitors) EvalMetric (class in lumin.nn.metrics.eval_metric) evaluate() (lumin.nn.metrics.class_eval.AMS method) (lumin.nn.metrics.class_eval.BinaryAccuracy method) (lumin.nn.metrics.class_eval.MultiAMS method) (lumin.nn.metrics.class_eval.RocAucScore method) (lumin.nn.metrics.eval_metric.EvalMetric method) (lumin.nn.metrics.reg_eval.RegAsProxyPull method) (lumin.nn.metrics.reg_eval.RegPull method) (lumin.nn.models.model.Model method)

evaluate_model() (lumin.nn.metrics.eval_metric.EvalMetric method) evaluate_preds() (lumin.nn.metrics.eval_metric.EvalMetric method) event_to_cartesian() (in module lumin.data_processing.hep_proc) export2onnx() (lumin.nn.ensemble.ensemble.Ensemble method) (lumin.nn.models.model.Model method) export2tfpb() (lumin.nn.ensemble.ensemble.Ensemble method) (lumin.nn.models.model.Model method)

F

feat_extractor() (lumin.nn.models.blocks.head.AutoExtractLorentzBoostNet method) (lumin.nn.models.blocks.head.LorentzBoostNet method) fit() (lumin.nn.models.model.Model method) fit_input_pipe() (in module lumin.data_processing.pre_proc) fit_output_pipe() (in module lumin.data_processing.pre_proc) fix_event_phi() (in module lumin.data_processing.hep_proc) fix_event_y() (in module lumin.data_processing.hep_proc) fix_event_z() (in module lumin.data_processing.hep_proc) fold2foldfile() (in module lumin.data_processing.file_proc) FoldYielder (class in lumin.nn.data.fold_yielder) forward() (lumin.nn.losses.advanced_losses.WeightedBinnedHuber method) (lumin.nn.losses.advanced_losses.WeightedFractionalBinnedHuber method) (lumin.nn.losses.advanced_losses.WeightedFractionalMSE method) (lumin.nn.losses.basic_weighted.WeightedCCE method) (lumin.nn.losses.basic_weighted.WeightedMAE method) (lumin.nn.losses.basic_weighted.WeightedMSE method) (lumin.nn.losses.hep_losses.SignificanceLoss method) (lumin.nn.models.blocks.body.FullyConnected method) (lumin.nn.models.blocks.body.IdentBody method) (lumin.nn.models.blocks.body.MultiBlock method) (lumin.nn.models.blocks.conv_blocks.AdaptiveAvgMaxConcatPool1d method) (lumin.nn.models.blocks.conv_blocks.Conv1DBlock method) (lumin.nn.models.blocks.conv_blocks.Res1DBlock method) (lumin.nn.models.blocks.conv_blocks.ResNeXt1DBlock method) (lumin.nn.models.blocks.conv_blocks.SEBlock1d method) (lumin.nn.models.blocks.endcap.AbsEndcap method) (lumin.nn.models.blocks.gnn_blocks.GraphCollapser method) (lumin.nn.models.blocks.gnn_blocks.GravNet method) (lumin.nn.models.blocks.gnn_blocks.GravNetLayer method) (lumin.nn.models.blocks.gnn_blocks.InteractionNet method) (lumin.nn.models.blocks.head.AbsConv1dHead method) (lumin.nn.models.blocks.head.CatEmbHead method) (lumin.nn.models.blocks.head.GNNHead method) (lumin.nn.models.blocks.head.LorentzBoostNet method) (lumin.nn.models.blocks.head.MultiHead method) (lumin.nn.models.blocks.head.RecurrentHead method) (lumin.nn.models.blocks.tail.ClassRegMulti method) (lumin.nn.models.blocks.tail.IdentTail method) (lumin.nn.models.layers.activations.Swish method) (lumin.nn.models.layers.batchnorms.LCBatchNorm1d method) (lumin.nn.models.layers.batchnorms.RunningBatchNorm1d method) (lumin.nn.models.layers.batchnorms.RunningBatchNorm2d method) (lumin.nn.models.layers.mish.Mish method) (lumin.nn.models.layers.self_attention.OffsetSelfAttention method) (lumin.nn.models.layers.self_attention.SelfAttention method)

ForwardHook (class in lumin.utils.misc) freeze_layers() (lumin.nn.models.model.Model method) from_fy() (lumin.nn.models.helpers.CatEmbedder class method) from_model_builder() (lumin.nn.models.model_builder.ModelBuilder class method) from_models() (lumin.nn.ensemble.ensemble.Ensemble class method) from_results() (lumin.nn.ensemble.ensemble.Ensemble class method) from_save() (lumin.nn.ensemble.ensemble.Ensemble class method) (lumin.nn.models.model.Model class method) FullyConnected (class in lumin.nn.models.blocks.body) func() (lumin.nn.models.blocks.endcap.AbsEndcap method)

G

get_body() (lumin.nn.models.model_builder.ModelBuilder method) get_column() (lumin.nn.data.fold_yielder.FoldYielder method) (lumin.nn.data.fold_yielder.TorchGeometricFoldYielder method) get_conv1d_block() (lumin.nn.models.blocks.head.AbsConv1dHead method) get_conv1d_res_block() (lumin.nn.models.blocks.head.AbsConv1dHead method) get_conv1d_resNeXt_block() (lumin.nn.models.blocks.head.AbsConv1dHead method) get_conv_layer() (lumin.nn.models.blocks.conv_blocks.Conv1DBlock method) get_data() (lumin.nn.data.fold_yielder.FoldYielder method) (lumin.nn.data.fold_yielder.TorchGeometricFoldYielder method) get_data_count() (lumin.nn.data.fold_yielder.FoldYielder method) get_df() (lumin.nn.callbacks.opt_callbacks.LRFinder method) (lumin.nn.data.fold_yielder.FoldYielder method) (lumin.nn.data.fold_yielder.TorchGeometricFoldYielder method) (lumin.nn.metrics.eval_metric.EvalMetric method) get_embeds() (lumin.nn.models.blocks.head.CatEmbHead method) get_ensemble_feat_importance() (in module lumin.nn.interpretation.features) get_feat_importance() (lumin.nn.ensemble.ensemble.Ensemble method) (lumin.nn.models.model.Model method) get_fold() (lumin.nn.data.fold_yielder.FoldYielder method) (lumin.nn.data.fold_yielder.HEPAugFoldYielder method) (lumin.nn.data.fold_yielder.TorchGeometricFoldYielder method) get_head() (lumin.nn.models.model_builder.ModelBuilder method) get_ignore() (lumin.nn.data.fold_yielder.FoldYielder method) get_inputs() (lumin.nn.data.batch_yielder.BatchYielder method) (lumin.nn.data.batch_yielder.TorchGeometricBatchYielder method) get_layers() (lumin.nn.models.blocks.head.AbsConv1dHead method) get_loss() (lumin.nn.callbacks.model_callbacks.SWA method) get_loss_history() (lumin.nn.callbacks.monitors.MetricLogger method) get_lr() (lumin.nn.models.model.Model method) get_metric() (lumin.nn.metrics.eval_metric.EvalMetric method) get_model() (lumin.nn.models.model_builder.ModelBuilder method) get_mom() (lumin.nn.models.model.Model method) get_moments() (in module lumin.utils.statistics) get_momentum() (in module lumin.data_processing.hep_proc) get_nn_feat_importance() (in module lumin.nn.interpretation.features)

get_opt_rf_params() (in module lumin.optimisation.hyper_param) get_out_size() (lumin.nn.models.blocks.body.FullyConnected method) (lumin.nn.models.blocks.body.IdentBody method) (lumin.nn.models.blocks.body.MultiBlock method) (lumin.nn.models.blocks.gnn_blocks.GraphCollapser method) (lumin.nn.models.blocks.gnn_blocks.GravNet method) (lumin.nn.models.blocks.gnn_blocks.GravNetLayer method) (lumin.nn.models.blocks.gnn_blocks.InteractionNet method) (lumin.nn.models.blocks.head.AbsConv1dHead method) (lumin.nn.models.blocks.head.CatEmbHead method) (lumin.nn.models.blocks.head.GNNHead method) (lumin.nn.models.blocks.head.LorentzBoostNet method) (lumin.nn.models.blocks.head.MultiHead method) (lumin.nn.models.blocks.head.RecurrentHead method) (lumin.nn.models.blocks.tail.ClassRegMulti method) (lumin.nn.models.blocks.tail.IdentTail method) (lumin.nn.models.layers.self_attention.SelfAttention method) (lumin.nn.models.model.Model method) (lumin.nn.models.model_builder.ModelBuilder method) get_padding() (lumin.nn.models.blocks.conv_blocks.Conv1DBlock static method) get_param_count() (lumin.nn.models.model.Model method) get_pre_proc_pipes() (in module lumin.data_processing.pre_proc) get_preds() (lumin.nn.callbacks.pred_handlers.PredHandler method) get_results() (lumin.nn.callbacks.monitors.MetricLogger method) get_rf_feat_importance() (in module lumin.optimisation.features) get_tail() (lumin.nn.models.model_builder.ModelBuilder method) get_test_fold() (lumin.nn.data.fold_yielder.HEPAugFoldYielder method) get_use_cat_feats() (lumin.nn.data.fold_yielder.FoldYielder method) get_use_cont_feats() (lumin.nn.data.fold_yielder.FoldYielder method) get_vecs() (in module lumin.data_processing.hep_proc) get_weights() (lumin.nn.models.model.Model method) GNNHead (class in lumin.nn.models.blocks.head) GradClip (class in lumin.nn.callbacks.loss_callbacks) GraphCollapser (class in lumin.nn.models.blocks.gnn_blocks) GravNet (class in lumin.nn.models.blocks.gnn_blocks) GravNetLayer (class in lumin.nn.models.blocks.gnn_blocks)

H

hard_identity() (in module lumin.utils.misc)

HEPAugFoldYielder (class in lumin.nn.data.fold_yielder) hook_fn() (lumin.utils.misc.ForwardHook method)

I

IdentBody (class in lumin.nn.models.blocks.body) IdentTail (class in lumin.nn.models.blocks.tail)

ids2unique() (in module lumin.utils.misc) InteractionNet (class in lumin.nn.models.blocks.gnn_blocks) is_partially() (in module lumin.utils.misc)

L

LCBatchNorm1d (class in lumin.nn.models.layers.batchnorms) load() (lumin.nn.ensemble.ensemble.Ensemble method) (lumin.nn.models.model.Model method) load_pretrained() (lumin.nn.models.model_builder.ModelBuilder method) load_trained_model() (lumin.nn.ensemble.ensemble.Ensemble static method) lookup_act() (in module lumin.nn.models.layers.activations) lookup_normal_init() (in module lumin.nn.models.initialisations) lookup_uniform_init() (in module lumin.nn.models.initialisations) LorentzBoostNet (class in lumin.nn.models.blocks.head) lr_find() (in module lumin.optimisation.hyper_param) LRFinder (class in lumin.nn.callbacks.opt_callbacks) LsuvInit (class in lumin.nn.callbacks.lsuv_init) lumin module lumin.data_processing module lumin.data_processing.file_proc module lumin.data_processing.hep_proc module lumin.data_processing.pre_proc module lumin.evaluation module lumin.evaluation.ams module lumin.inference module lumin.inference.summary_stat module lumin.nn module lumin.nn.callbacks module lumin.nn.callbacks.adversarial_callbacks module lumin.nn.callbacks.callback module lumin.nn.callbacks.cyclic_callbacks module lumin.nn.callbacks.data_callbacks module lumin.nn.callbacks.loss_callbacks module lumin.nn.callbacks.lsuv_init module lumin.nn.callbacks.model_callbacks module lumin.nn.callbacks.monitors module lumin.nn.callbacks.opt_callbacks module lumin.nn.callbacks.pred_handlers module lumin.nn.data module lumin.nn.data.batch_yielder module lumin.nn.data.fold_yielder module lumin.nn.ensemble module lumin.nn.ensemble.ensemble module lumin.nn.interpretation module lumin.nn.interpretation.features module lumin.nn.losses module lumin.nn.losses.advanced_losses module lumin.nn.losses.basic_weighted module lumin.nn.losses.hep_losses module

lumin.nn.metrics module lumin.nn.metrics.class_eval module lumin.nn.metrics.eval_metric module lumin.nn.metrics.reg_eval module lumin.nn.models module lumin.nn.models.blocks module lumin.nn.models.blocks.body module lumin.nn.models.blocks.conv_blocks module lumin.nn.models.blocks.endcap module lumin.nn.models.blocks.gnn_blocks module lumin.nn.models.blocks.head module lumin.nn.models.blocks.tail module lumin.nn.models.helpers module lumin.nn.models.initialisations module lumin.nn.models.layers module lumin.nn.models.layers.activations module lumin.nn.models.layers.batchnorms module lumin.nn.models.layers.mish module lumin.nn.models.layers.self_attention module lumin.nn.models.model module lumin.nn.models.model_builder module lumin.nn.training module lumin.nn.training.train module lumin.optimisation module lumin.optimisation.features module lumin.optimisation.hyper_param module lumin.optimisation.threshold module lumin.plotting module lumin.plotting.data_viewing module lumin.plotting.interpretation module lumin.plotting.plot_settings module lumin.plotting.results module lumin.plotting.training module lumin.utils module lumin.utils.data module lumin.utils.misc module lumin.utils.multiprocessing module lumin.utils.statistics module

M

MetricLogger (class in lumin.nn.callbacks.monitors) Mish (class in lumin.nn.models.layers.mish) Model (class in lumin.nn.models.model) ModelBuilder (class in lumin.nn.models.model_builder) module lumin lumin.data_processing lumin.data_processing.file_proc lumin.data_processing.hep_proc lumin.data_processing.pre_proc lumin.evaluation lumin.evaluation.ams lumin.inference lumin.inference.summary_stat lumin.nn lumin.nn.callbacks lumin.nn.callbacks.adversarial_callbacks lumin.nn.callbacks.callback lumin.nn.callbacks.cyclic_callbacks lumin.nn.callbacks.data_callbacks lumin.nn.callbacks.loss_callbacks lumin.nn.callbacks.lsuv_init lumin.nn.callbacks.model_callbacks lumin.nn.callbacks.monitors lumin.nn.callbacks.opt_callbacks lumin.nn.callbacks.pred_handlers lumin.nn.data lumin.nn.data.batch_yielder lumin.nn.data.fold_yielder lumin.nn.ensemble lumin.nn.ensemble.ensemble lumin.nn.interpretation lumin.nn.interpretation.features lumin.nn.losses lumin.nn.losses.advanced_losses lumin.nn.losses.basic_weighted lumin.nn.losses.hep_losses lumin.nn.metrics lumin.nn.metrics.class_eval lumin.nn.metrics.eval_metric lumin.nn.metrics.reg_eval lumin.nn.models lumin.nn.models.blocks lumin.nn.models.blocks.body lumin.nn.models.blocks.conv_blocks lumin.nn.models.blocks.endcap lumin.nn.models.blocks.gnn_blocks lumin.nn.models.blocks.head lumin.nn.models.blocks.tail lumin.nn.models.helpers lumin.nn.models.initialisations lumin.nn.models.layers lumin.nn.models.layers.activations lumin.nn.models.layers.batchnorms lumin.nn.models.layers.mish lumin.nn.models.layers.self_attention lumin.nn.models.model lumin.nn.models.model_builder lumin.nn.training lumin.nn.training.train lumin.optimisation lumin.optimisation.features lumin.optimisation.hyper_param lumin.optimisation.threshold lumin.plotting lumin.plotting.data_viewing lumin.plotting.interpretation lumin.plotting.plot_settings lumin.plotting.results lumin.plotting.training lumin.utils lumin.utils.data lumin.utils.misc lumin.utils.multiprocessing lumin.utils.statistics

mp_run() (in module lumin.utils.multiprocessing) MultiAMS (class in lumin.nn.metrics.class_eval) MultiBlock (class in lumin.nn.models.blocks.body) MultiHead (class in lumin.nn.models.blocks.head)

N

NodePredictor (class in lumin.nn.models.blocks.gnn_blocks)

O

OffsetSelfAttention (class in lumin.nn.models.layers.self_attention) on_backwards_end() (lumin.nn.callbacks.loss_callbacks.GradClip method) on_batch_begin() (lumin.nn.callbacks.adversarial_callbacks.PivotTraining method) (lumin.nn.callbacks.cyclic_callbacks.AbsCyclicCallback method) (lumin.nn.callbacks.cyclic_callbacks.CycleStep method) (lumin.nn.callbacks.cyclic_callbacks.OneCycle method) on_batch_end() (lumin.nn.callbacks.cyclic_callbacks.AbsCyclicCallback method) (lumin.nn.callbacks.monitors.MetricLogger method) (lumin.nn.callbacks.opt_callbacks.LRFinder method) on_epoch_begin() (lumin.nn.callbacks.adversarial_callbacks.PivotTraining method) (lumin.nn.callbacks.cyclic_callbacks.AbsCyclicCallback method) (lumin.nn.callbacks.cyclic_callbacks.CycleStep method) (lumin.nn.callbacks.model_callbacks.SWA method) (lumin.nn.callbacks.monitors.MetricLogger method) (lumin.nn.callbacks.opt_callbacks.LRFinder method) (lumin.nn.metrics.eval_metric.EvalMetric method) (lumin.nn.metrics.eval_metric.TorchGeometricEvalMetric method) on_epoch_end() (lumin.nn.callbacks.data_callbacks.TargReplace method) (lumin.nn.callbacks.model_callbacks.SWA method) (lumin.nn.callbacks.monitors.EarlyStopping method) (lumin.nn.callbacks.monitors.EpochSaver method) (lumin.nn.callbacks.monitors.MetricLogger method) (lumin.nn.callbacks.monitors.SaveBest method) (lumin.nn.metrics.eval_metric.EvalMetric method) (lumin.nn.metrics.eval_metric.TorchGeometricEvalMetric method) on_fold_begin() (lumin.nn.callbacks.adversarial_callbacks.PivotTraining method) (lumin.nn.callbacks.data_callbacks.BinaryLabelSmooth method) (lumin.nn.callbacks.data_callbacks.BootstrapResample method) (lumin.nn.callbacks.data_callbacks.TargReplace method) (lumin.nn.callbacks.lsuv_init.LsuvInit method) (lumin.nn.callbacks.monitors.MetricLogger method)

on_fold_end() (lumin.nn.callbacks.monitors.MetricLogger method) on_forwards_end() (lumin.nn.callbacks.adversarial_callbacks.PivotTraining method) (lumin.nn.callbacks.pred_handlers.PredHandler method) (lumin.nn.metrics.eval_metric.EvalMetric method) (lumin.nn.metrics.eval_metric.TorchGeometricEvalMetric method) on_pred_begin() (lumin.nn.callbacks.callback.Callback method) (lumin.nn.callbacks.data_callbacks.ParametrisedPrediction method) (lumin.nn.callbacks.pred_handlers.PredHandler method) on_pred_end() (lumin.nn.callbacks.pred_handlers.PredHandler method) on_train_begin() (lumin.nn.callbacks.adversarial_callbacks.PivotTraining method) (lumin.nn.callbacks.callback.Callback method) (lumin.nn.callbacks.cyclic_callbacks.AbsCyclicCallback method) (lumin.nn.callbacks.cyclic_callbacks.CycleStep method) (lumin.nn.callbacks.data_callbacks.BootstrapResample method) (lumin.nn.callbacks.lsuv_init.LsuvInit method) (lumin.nn.callbacks.model_callbacks.SWA method) (lumin.nn.callbacks.monitors.EarlyStopping method) (lumin.nn.callbacks.monitors.EpochSaver method) (lumin.nn.callbacks.monitors.MetricLogger method) (lumin.nn.callbacks.monitors.SaveBest method) (lumin.nn.callbacks.opt_callbacks.LRFinder method) (lumin.nn.metrics.eval_metric.EvalMetric method) on_train_end() (lumin.nn.callbacks.adversarial_callbacks.PivotTraining method) (lumin.nn.callbacks.cyclic_callbacks.CycleStep method) (lumin.nn.callbacks.monitors.SaveBest method) OneCycle (class in lumin.nn.callbacks.cyclic_callbacks)

P

ParametrisedPrediction (class in lumin.nn.callbacks.data_callbacks) PivotTraining (class in lumin.nn.callbacks.adversarial_callbacks) plot() (lumin.nn.callbacks.cyclic_callbacks.AbsCyclicCallback method) (lumin.nn.callbacks.cyclic_callbacks.CycleStep method) (lumin.nn.callbacks.cyclic_callbacks.OneCycle method) (lumin.nn.callbacks.opt_callbacks.LRFinder method) plot_1d_partial_dependence() (in module lumin.plotting.interpretation) plot_2d_partial_dependence() (in module lumin.plotting.interpretation) plot_binary_class_pred() (in module lumin.plotting.results) plot_binary_sample_feat() (in module lumin.plotting.data_viewing) plot_bottleneck_weighted_inputs() (in module lumin.plotting.interpretation) plot_embedding() (in module lumin.plotting.interpretation) plot_embeds() (lumin.nn.models.blocks.head.CatEmbHead method) plot_feat() (in module lumin.plotting.data_viewing) plot_importance() (in module lumin.plotting.interpretation) plot_kdes_from_bs() (in module lumin.plotting.data_viewing)

plot_lr() (lumin.nn.callbacks.opt_callbacks.LRFinder method) plot_lr_finders() (in module lumin.plotting.training) plot_multibody_weighted_outputs() (in module lumin.plotting.interpretation) plot_rank_order_dendrogram() (in module lumin.plotting.data_viewing) plot_roc() (in module lumin.plotting.results) plot_sample_pred() (in module lumin.plotting.results) plot_train_history() (in module lumin.plotting.training) PlotSettings (class in lumin.plotting.plot_settings) potential() (lumin.nn.models.blocks.gnn_blocks.GravNetLayer static method) PredHandler (class in lumin.nn.callbacks.pred_handlers) predict() (lumin.nn.ensemble.ensemble.Ensemble method) (lumin.nn.models.blocks.endcap.AbsEndcap method) (lumin.nn.models.model.Model method) print_losses() (lumin.nn.callbacks.monitors.MetricLogger method) proc_cats() (in module lumin.data_processing.pre_proc) proc_event() (in module lumin.data_processing.hep_proc)

R

RecurrentHead (class in lumin.nn.models.blocks.head) RegAsProxyPull (class in lumin.nn.metrics.reg_eval) RegPull (class in lumin.nn.metrics.reg_eval) remove() (lumin.utils.misc.ForwardHook method) repeated_rf_rank_features() (in module lumin.optimisation.features) Res1DBlock (class in lumin.nn.models.blocks.conv_blocks) ResNeXt1DBlock (class in lumin.nn.models.blocks.conv_blocks)

rf_check_feat_removal() (in module lumin.optimisation.features) rf_rank_features() (in module lumin.optimisation.features) RocAucScore (class in lumin.nn.metrics.class_eval) row_wise (lumin.nn.models.blocks.gnn_blocks.GravNet attribute) (lumin.nn.models.blocks.gnn_blocks.InteractionNet attribute) RunningBatchNorm1d (class in lumin.nn.models.layers.batchnorms) RunningBatchNorm2d (class in lumin.nn.models.layers.batchnorms) RunningBatchNorm3d (class in lumin.nn.models.layers.batchnorms)

S

save() (lumin.nn.ensemble.ensemble.Ensemble method) (lumin.nn.models.model.Model method) save_embeds() (lumin.nn.models.blocks.head.CatEmbHead method) save_fold_pred() (lumin.nn.data.fold_yielder.FoldYielder method) (lumin.nn.data.fold_yielder.TorchGeometricFoldYielder method) save_to_grp() (in module lumin.data_processing.file_proc) SaveBest (class in lumin.nn.callbacks.monitors) SEBlock1d (class in lumin.nn.models.blocks.conv_blocks) SEBlock2d (class in lumin.nn.models.blocks.conv_blocks) SEBlock3d (class in lumin.nn.models.blocks.conv_blocks) SelfAttention (class in lumin.nn.models.layers.self_attention) set_input_mask() (lumin.nn.models.model.Model method) set_layers() (lumin.nn.models.blocks.conv_blocks.Conv1DBlock method) (lumin.nn.models.blocks.conv_blocks.Res1DBlock method) (lumin.nn.models.blocks.conv_blocks.ResNeXt1DBlock method)

set_lr() (lumin.nn.models.model.Model method) (lumin.nn.models.model_builder.ModelBuilder method) set_model() (lumin.nn.callbacks.callback.Callback method) set_mom() (lumin.nn.models.model.Model method) set_plot_settings() (lumin.nn.callbacks.callback.Callback method) set_weights() (lumin.nn.models.model.Model method) SignificanceLoss (class in lumin.nn.losses.hep_losses) str2bool() (in module lumin.utils.misc) str2sz() (lumin.plotting.plot_settings.PlotSettings method) subsample_df() (in module lumin.utils.misc) SWA (class in lumin.nn.callbacks.model_callbacks) Swish (class in lumin.nn.models.layers.activations)

T

TargReplace (class in lumin.nn.callbacks.data_callbacks) to_binary_class() (in module lumin.utils.misc) to_cartesian() (in module lumin.data_processing.hep_proc) to_device() (in module lumin.utils.misc) to_np() (in module lumin.utils.misc) to_pt_eta_phi() (in module lumin.data_processing.hep_proc)

to_tensor() (in module lumin.utils.misc) TorchGeometricBatchYielder (class in lumin.nn.data.batch_yielder) TorchGeometricEvalMetric (class in lumin.nn.metrics.eval_metric) TorchGeometricFoldYielder (class in lumin.nn.data.fold_yielder) train_models() (in module lumin.nn.training.train) twist() (in module lumin.data_processing.hep_proc)

U

uncert_round() (in module lumin.utils.statistics) unfreeze_layers() (lumin.nn.models.model.Model method)

update_plot() (lumin.nn.callbacks.monitors.MetricLogger method) update_stats() (lumin.nn.models.layers.batchnorms.RunningBatchNorm1d method)

W

WeightedBinnedHuber (class in lumin.nn.losses.advanced_losses) WeightedCCE (class in lumin.nn.losses.basic_weighted) WeightedFractionalBinnedHuber (class in lumin.nn.losses.advanced_losses)

WeightedFractionalMSE (class in lumin.nn.losses.advanced_losses) WeightedMAE (class in lumin.nn.losses.basic_weighted) WeightedMSE (class in lumin.nn.losses.basic_weighted)