Index

A

activation_extraction (bead.src.utils.ggl.Config attribute) add_sig_bkg_label() (in module bead.src.utils.helper)

AE (class in bead.src.models.models) AE_Dropout_BN (class in bead.src.models.models) append_to_hdf5() (in module bead.src.utils.conversion)

B

BaseLoss (class in bead.src.utils.loss) batch_construct_orthogonal() (bead.src.models.models.HouseholderSylvester_ConvVAE method) (bead.src.models.models.OrthogonalSylvester_ConvVAE method) batch_size (bead.src.utils.ggl.Config attribute) bead module bead.bead module bead.src module bead.src.models module bead.src.models.flows module bead.src.models.layers module bead.src.models.models module bead.src.trainers module bead.src.trainers.inference module bead.src.trainers.training module

bead.src.utils module bead.src.utils.conversion module bead.src.utils.data_processing module bead.src.utils.diagnostics module bead.src.utils.ggl module bead.src.utils.helper module bead.src.utils.loss module bead.src.utils.normalization module bead.src.utils.plotting module best_loss (bead.src.utils.helper.EarlyStopping attribute) BinaryCrossEntropyLoss (class in bead.src.utils.loss) build_mask() (bead.src.models.layers.MaskedConv2d method) (bead.src.models.layers.MaskedLinear method)

C

c_profile() (in module bead.src.utils.diagnostics) calculate() (bead.src.utils.loss.BaseLoss method) (bead.src.utils.loss.BinaryCrossEntropyLoss method) (bead.src.utils.loss.KLDivergenceLoss method) (bead.src.utils.loss.L1Regularization method) (bead.src.utils.loss.L2Regularization method) (bead.src.utils.loss.ReconstructionLoss method) (bead.src.utils.loss.SupervisedContrastiveLoss method) (bead.src.utils.loss.VAEFlowLoss method) (bead.src.utils.loss.VAEFlowLossEMD method) (bead.src.utils.loss.VAEFlowLossL1 method) (bead.src.utils.loss.VAEFlowLossL2 method) (bead.src.utils.loss.VAEFlowSupConLoss method) (bead.src.utils.loss.VAELoss method) (bead.src.utils.loss.VAELossEMD method) (bead.src.utils.loss.VAELossL1 method) (bead.src.utils.loss.VAELossL2 method) (bead.src.utils.loss.VAESupConLoss method) (bead.src.utils.loss.WassersteinLoss method) calculate_in_shape() (in module bead.src.utils.helper) calculate_jet_properties_numba() (in module bead.src.utils.conversion)

call_forward() (in module bead.src.utils.helper) ChainedScaler (class in bead.src.utils.helper) chunk_size (bead.src.utils.ggl.Config attribute) CNN_Flow (class in bead.src.models.flows) CNN_Flow_Layer (class in bead.src.models.layers) compute_alpha_laplace() (bead.src.utils.loss.KLDivergenceLoss method) Config (class in bead.src.utils.ggl) contrastive_temperature (bead.src.utils.ggl.Config attribute) contrastive_weight (bead.src.utils.ggl.Config attribute) ConvAE (class in bead.src.models.models) convert_csv() (in module bead.src.utils.ggl) convert_csv_to_hdf5_npy_parallel() (in module bead.src.utils.conversion) convert_to_tensor() (in module bead.src.utils.helper) ConvFlow_ConvVAE (class in bead.src.models.models) ConvVAE (class in bead.src.models.models) counter (bead.src.utils.helper.EarlyStopping attribute) create_datasets() (in module bead.src.utils.helper) create_default_config() (in module bead.src.utils.ggl) create_new_project() (in module bead.src.utils.ggl) csv_chunk_generator() (in module bead.src.utils.conversion) CustomDataset (class in bead.src.utils.helper)

D

data (bead.src.utils.helper.CustomDataset attribute) data_label_split() (in module bead.src.utils.helper) dec_bn() (bead.src.models.models.AE_Dropout_BN method) decode() (bead.src.models.models.AE method) (bead.src.models.models.ConvAE method) (bead.src.models.models.ConvVAE method) (bead.src.models.models.Dirichlet_ConvVAE method) decoder() (bead.src.models.models.TransformerAE method) der_h() (bead.src.models.flows.Planar method) (bead.src.models.flows.Sylvester method) (bead.src.models.flows.TriangularSylvester method)

der_tanh() (bead.src.models.flows.Sylvester method) (bead.src.models.flows.TriangularSylvester method) detach_device() (in module bead.src.utils.helper) detach_hooks() (bead.src.models.models.AE method) deterministic_algorithm (bead.src.utils.ggl.Config attribute) dict_to_square_matrix() (in module bead.src.utils.diagnostics) Dilation_Block (class in bead.src.models.layers) Dirichlet_ConvVAE (class in bead.src.models.models) DVAEFlowLoss (class in bead.src.utils.loss) DVAELoss (class in bead.src.utils.loss)

E

early_stop (bead.src.utils.helper.EarlyStopping attribute) early_stoppin_patience (bead.src.utils.ggl.Config attribute) early_stopping (bead.src.utils.ggl.Config attribute) EarlyStopping (class in bead.src.utils.helper) enc_bn() (bead.src.models.models.AE_Dropout_BN method) encode() (bead.src.models.models.AE method) (bead.src.models.models.ConvAE method) (bead.src.models.models.ConvVAE method) (bead.src.models.models.Dirichlet_ConvVAE method) (bead.src.models.models.IAF_ConvVAE method)

encoder() (bead.src.models.models.TransformerAE method) epochs (bead.src.utils.ggl.Config attribute)

F

FCNN (class in bead.src.models.layers) file_type (bead.src.utils.ggl.Config attribute) fit() (bead.src.utils.helper.ChainedScaler method) (bead.src.utils.helper.L2Normalizer method) (bead.src.utils.helper.Log1pScaler method) (bead.src.utils.helper.SinCosTransformer method) (in module bead.src.trainers.training) forward() (bead.src.models.flows.CNN_Flow method) (bead.src.models.flows.IAF method) (bead.src.models.flows.NSF_AR method) (bead.src.models.flows.Planar method) (bead.src.models.flows.Sylvester method) (bead.src.models.flows.TriangularSylvester method) (bead.src.models.layers.CNN_Flow_Layer method) (bead.src.models.layers.Dilation_Block method) (bead.src.models.layers.FCNN method) (bead.src.models.layers.GraphConvolution method) (bead.src.models.layers.Identity method) (bead.src.models.layers.MaskedConv2d method) (bead.src.models.layers.MaskedLinear method) (bead.src.models.models.AE method) (bead.src.models.models.AE_Dropout_BN method) (bead.src.models.models.ConvAE method) (bead.src.models.models.ConvFlow_ConvVAE method) (bead.src.models.models.ConvVAE method) (bead.src.models.models.Dirichlet_ConvVAE method) (bead.src.models.models.HouseholderSylvester_ConvVAE method) (bead.src.models.models.IAF_ConvVAE method) (bead.src.models.models.NSFAR_ConvVAE method) (bead.src.models.models.OrthogonalSylvester_ConvVAE method) (bead.src.models.models.Planar_ConvVAE method) (bead.src.models.models.TransformerAE method) (bead.src.models.models.TriangularSylvester_ConvVAE method)

G

get_activations() (bead.src.models.models.AE method) get_arguments() (in module bead.src.utils.ggl) get_device() (in module bead.src.utils.helper) get_hook() (bead.src.models.models.AE method)

get_layers() (bead.src.models.models.AE method) get_loss() (in module bead.src.utils.helper) get_mean_node_activations() (in module bead.src.utils.diagnostics) get_optimizer() (in module bead.src.utils.helper) GraphConvolution (class in bead.src.models.layers)

H

HouseholderSylvester_ConvVAE (class in bead.src.models.models)

I

IAF (class in bead.src.models.flows) IAF_ConvVAE (class in bead.src.models.models) Identity (class in bead.src.models.layers) infer() (in module bead.src.trainers.inference) input_features (bead.src.utils.ggl.Config attribute) input_level (bead.src.utils.ggl.Config attribute) intermittent_model_saving (bead.src.utils.ggl.Config attribute) intermittent_saving_patience (bead.src.utils.ggl.Config attribute)

inverse_transform() (bead.src.utils.helper.ChainedScaler method) (bead.src.utils.helper.L2Normalizer method) (bead.src.utils.helper.Log1pScaler method) (bead.src.utils.helper.SinCosTransformer method) invert_normalizations (bead.src.utils.ggl.Config attribute) invert_normalize_constit_pj_custom() (in module bead.src.utils.normalization) invert_normalize_data() (in module bead.src.utils.helper) invert_normalize_jet_pj_custom() (in module bead.src.utils.normalization)

K

KLDivergenceLoss (class in bead.src.utils.loss)

L

L1Regularization (class in bead.src.utils.loss) L2Normalizer (class in bead.src.utils.helper) L2Regularization (class in bead.src.utils.loss) labels (bead.src.utils.helper.CustomDataset attribute) latent_space_plot_style (bead.src.utils.ggl.Config attribute) latent_space_size (bead.src.utils.ggl.Config attribute) load_augment_tensors() (in module bead.src.utils.helper) load_data() (in module bead.src.utils.data_processing)

load_model() (in module bead.src.utils.helper) load_tensors() (in module bead.src.utils.helper) Log1pScaler (class in bead.src.utils.helper) loss_function (bead.src.utils.ggl.Config attribute) lr (bead.src.utils.ggl.Config attribute) lr_scheduler (bead.src.utils.ggl.Config attribute) (bead.src.utils.helper.LRScheduler attribute) lr_scheduler_patience (bead.src.utils.ggl.Config attribute) LRScheduler (class in bead.src.utils.helper)

M

main() (in module bead.bead) MaskedConv2d (class in bead.src.models.layers) MaskedLinear (class in bead.src.models.layers) min_delta (bead.src.utils.ggl.Config attribute) model_init (bead.src.utils.ggl.Config attribute) model_init() (in module bead.src.utils.helper) model_name (bead.src.utils.ggl.Config attribute) module bead bead.bead bead.src bead.src.models bead.src.models.flows bead.src.models.layers bead.src.models.models bead.src.trainers bead.src.trainers.inference bead.src.trainers.training bead.src.utils bead.src.utils.conversion bead.src.utils.data_processing bead.src.utils.diagnostics bead.src.utils.ggl bead.src.utils.helper bead.src.utils.loss bead.src.utils.normalization bead.src.utils.plotting

N

nap_diagnose() (in module bead.src.utils.diagnostics) normalizations (bead.src.utils.ggl.Config attribute) normalize_constit_pj_custom() (in module bead.src.utils.normalization) normalize_data() (in module bead.src.utils.helper) normalize_jet_pj_custom() (in module bead.src.utils.normalization)

NSF_AR (class in bead.src.models.flows) NSFAR_ConvVAE (class in bead.src.models.models) num_constits (bead.src.utils.ggl.Config attribute) num_jets (bead.src.utils.ggl.Config attribute) numpy_to_tensor() (in module bead.src.utils.helper)

O

optimizer (bead.src.utils.ggl.Config attribute) OrthogonalSylvester_ConvVAE (class in bead.src.models.models) overlay_roc (bead.src.utils.ggl.Config attribute)

overlay_roc_filename (bead.src.utils.ggl.Config attribute) overlay_roc_projects (bead.src.utils.ggl.Config attribute) overlay_roc_save_location (bead.src.utils.ggl.Config attribute)

P

parallel_workers (bead.src.utils.ggl.Config attribute) Planar (class in bead.src.models.flows) Planar_ConvVAE (class in bead.src.models.models) plot() (in module bead.src.utils.diagnostics) plot_latent_variables() (in module bead.src.utils.plotting) plot_losses() (in module bead.src.utils.plotting) plot_mu_logvar() (in module bead.src.utils.plotting)

plot_roc_curve() (in module bead.src.utils.plotting) prepare_inputs() (in module bead.src.utils.ggl) preproc_inputs() (in module bead.src.utils.data_processing) process_and_save_tensors() (in module bead.src.utils.data_processing) process_chunk() (in module bead.src.utils.conversion) process_event() (in module bead.src.utils.conversion) project_name (bead.src.utils.ggl.Config attribute) pytorch_profile() (in module bead.src.utils.diagnostics)

R

ReconstructionLoss (class in bead.src.utils.loss) reduce_dim_subsampled() (in module bead.src.utils.plotting) reg_param (bead.src.utils.ggl.Config attribute) reparameterize() (bead.src.models.models.ConvVAE method) (bead.src.models.models.Dirichlet_ConvVAE method) reset_parameters() (bead.src.models.flows.NSF_AR method) (bead.src.models.layers.GraphConvolution method) (bead.src.models.layers.MaskedConv2d method) (bead.src.models.layers.MaskedLinear method)

RQS() (in module bead.src.models.layers) run_diagnostics() (in module bead.src.utils.ggl) run_full_chain() (in module bead.src.utils.ggl) run_inference() (in module bead.src.utils.ggl) run_plots() (in module bead.src.utils.ggl) run_training() (in module bead.src.utils.ggl)

S

save_loss_components() (in module bead.src.utils.helper) save_model() (in module bead.src.utils.helper) searchsorted() (in module bead.src.models.layers) seed_worker() (in module bead.src.trainers.inference) (in module bead.src.trainers.training) select_features() (in module bead.src.utils.helper) select_top_jets_and_constituents() (in module bead.src.utils.data_processing)

separate_model_saving (bead.src.utils.ggl.Config attribute) SinCosTransformer (class in bead.src.utils.helper) store_hooks() (bead.src.models.models.AE method) subsample_plot (bead.src.utils.ggl.Config attribute) subsample_size (bead.src.utils.ggl.Config attribute) SupervisedContrastiveLoss (class in bead.src.utils.loss) Sylvester (class in bead.src.models.flows)

T

train() (in module bead.src.trainers.training) train_size (bead.src.utils.ggl.Config attribute) train_val_split() (in module bead.src.utils.helper) transform() (bead.src.utils.helper.ChainedScaler method) (bead.src.utils.helper.L2Normalizer method) (bead.src.utils.helper.Log1pScaler method) (bead.src.utils.helper.SinCosTransformer method)

TransformerAE (class in bead.src.models.models) TriangularSylvester (class in bead.src.models.flows) TriangularSylvester_ConvVAE (class in bead.src.models.models)

U

unconstrained_RQS() (in module bead.src.models.layers)

use_amp (bead.src.utils.ggl.Config attribute) use_ddp (bead.src.utils.ggl.Config attribute)

V

VAEFlowLoss (class in bead.src.utils.loss) VAEFlowLossEMD (class in bead.src.utils.loss) VAEFlowLossL1 (class in bead.src.utils.loss) VAEFlowLossL2 (class in bead.src.utils.loss) VAEFlowSupConLoss (class in bead.src.utils.loss)

VAELoss (class in bead.src.utils.loss) VAELossEMD (class in bead.src.utils.loss) VAELossL1 (class in bead.src.utils.loss) VAELossL2 (class in bead.src.utils.loss) VAESupConLoss (class in bead.src.utils.loss) validate() (in module bead.src.trainers.training)

W

WassersteinLoss (class in bead.src.utils.loss)

workspace_name (bead.src.utils.ggl.Config attribute)