bead.src package

Subpackages

• bead.src.models package
  • Submodules
  • bead.src.models.flows module
    • CNN_Flow
      • CNN_Flow.forward()
    • IAF
      • IAF.forward()
    • NSF_AR
      • NSF_AR.forward()
      • NSF_AR.reset_parameters()
    • Planar
      • Planar.der_h()
      • Planar.forward()
    • Sylvester
      • Sylvester.der_h()
      • Sylvester.der_tanh()
      • Sylvester.forward()
    • TriangularSylvester
      • TriangularSylvester.der_h()
      • TriangularSylvester.der_tanh()
      • TriangularSylvester.forward()
  • bead.src.models.layers module
    • CNN_Flow_Layer
      • CNN_Flow_Layer.forward()
    • Dilation_Block
      • Dilation_Block.forward()
    • FCNN
      • FCNN.forward()
    • GraphConvolution
      • GraphConvolution.forward()
      • GraphConvolution.reset_parameters()
    • Identity
      • Identity.forward()
    • MaskedConv2d
      • MaskedConv2d.build_mask()
      • MaskedConv2d.forward()
      • MaskedConv2d.reset_parameters()
    • MaskedLinear
      • MaskedLinear.build_mask()
      • MaskedLinear.forward()
      • MaskedLinear.reset_parameters()
    • RQS()
    • searchsorted()
    • unconstrained_RQS()
  • bead.src.models.models module
    • AE
      • AE.decode()
      • AE.detach_hooks()
      • AE.encode()
      • AE.forward()
      • AE.get_activations()
      • AE.get_hook()
      • AE.get_layers()
      • AE.store_hooks()
    • AE_Dropout_BN
      • AE_Dropout_BN.dec_bn()
      • AE_Dropout_BN.enc_bn()
      • AE_Dropout_BN.forward()
    • ConvAE
      • ConvAE.decode()
      • ConvAE.encode()
      • ConvAE.forward()
    • ConvFlow_ConvVAE
      • ConvFlow_ConvVAE.forward()
    • ConvVAE
      • ConvVAE.decode()
      • ConvVAE.encode()
      • ConvVAE.forward()
      • ConvVAE.reparameterize()
    • Dirichlet_ConvVAE
      • Dirichlet_ConvVAE.decode()
      • Dirichlet_ConvVAE.encode()
      • Dirichlet_ConvVAE.forward()
      • Dirichlet_ConvVAE.reparameterize()
    • HouseholderSylvester_ConvVAE
      • HouseholderSylvester_ConvVAE.batch_construct_orthogonal()
      • HouseholderSylvester_ConvVAE.forward()
    • IAF_ConvVAE
      • IAF_ConvVAE.encode()
      • IAF_ConvVAE.forward()
    • NSFAR_ConvVAE
      • NSFAR_ConvVAE.forward()
    • OrthogonalSylvester_ConvVAE
      • OrthogonalSylvester_ConvVAE.batch_construct_orthogonal()
      • OrthogonalSylvester_ConvVAE.forward()
    • Planar_ConvVAE
      • Planar_ConvVAE.forward()
    • TransformerAE
      • TransformerAE.decoder()
      • TransformerAE.encoder()
      • TransformerAE.forward()
    • TriangularSylvester_ConvVAE
      • TriangularSylvester_ConvVAE.forward()
  • Module contents
• bead.src.trainers package
  • Submodules
  • bead.src.trainers.inference module
    • infer()
    • seed_worker()
  • bead.src.trainers.training module
    • fit()
    • seed_worker()
    • train()
    • validate()
  • Module contents
• bead.src.utils package
  • Submodules
  • bead.src.utils.conversion module
    • append_to_hdf5()
    • calculate_jet_properties_numba()
    • convert_csv_to_hdf5_npy_parallel()
    • csv_chunk_generator()
    • process_chunk()
    • process_event()
  • bead.src.utils.data_processing module
    • load_data()
    • preproc_inputs()
    • process_and_save_tensors()
    • select_top_jets_and_constituents()
  • bead.src.utils.diagnostics module
    • c_profile()
    • dict_to_square_matrix()
    • get_mean_node_activations()
    • nap_diagnose()
    • plot()
    • pytorch_profile()
  • bead.src.utils.ggl module
    • Config
      • Config.activation_extraction
      • Config.batch_size
      • Config.chunk_size
      • Config.contrastive_temperature
      • Config.contrastive_weight
      • Config.deterministic_algorithm
      • Config.early_stoppin_patience
      • Config.early_stopping
      • Config.epochs
      • Config.file_type
      • Config.input_features
      • Config.input_level
      • Config.intermittent_model_saving
      • Config.intermittent_saving_patience
      • Config.invert_normalizations
      • Config.latent_space_plot_style
      • Config.latent_space_size
      • Config.loss_function
      • Config.lr
      • Config.lr_scheduler
      • Config.lr_scheduler_patience
      • Config.min_delta
      • Config.model_init
      • Config.model_name
      • Config.normalizations
      • Config.num_constits
      • Config.num_jets
      • Config.optimizer
      • Config.overlay_roc
      • Config.overlay_roc_filename
      • Config.overlay_roc_projects
      • Config.overlay_roc_save_location
      • Config.parallel_workers
      • Config.project_name
      • Config.reg_param
      • Config.separate_model_saving
      • Config.subsample_plot
      • Config.subsample_size
      • Config.train_size
      • Config.use_amp
      • Config.use_ddp
      • Config.workspace_name
    • convert_csv()
    • create_default_config()
    • create_new_project()
    • get_arguments()
    • prepare_inputs()
    • run_diagnostics()
    • run_full_chain()
    • run_inference()
    • run_plots()
    • run_training()
  • bead.src.utils.helper module
    • ChainedScaler
      • ChainedScaler.fit()
      • ChainedScaler.inverse_transform()
      • ChainedScaler.transform()
    • CustomDataset
      • CustomDataset.data
      • CustomDataset.labels
    • EarlyStopping
      • EarlyStopping.counter
      • EarlyStopping.best_loss
      • EarlyStopping.early_stop
    • L2Normalizer
      • L2Normalizer.fit()
      • L2Normalizer.inverse_transform()
      • L2Normalizer.transform()
    • LRScheduler
      • LRScheduler.lr_scheduler
    • Log1pScaler
      • Log1pScaler.fit()
      • Log1pScaler.inverse_transform()
      • Log1pScaler.transform()
    • SinCosTransformer
      • SinCosTransformer.fit()
      • SinCosTransformer.inverse_transform()
      • SinCosTransformer.transform()
    • add_sig_bkg_label()
    • calculate_in_shape()
    • call_forward()
    • convert_to_tensor()
    • create_datasets()
    • data_label_split()
    • detach_device()
    • get_device()
    • get_loss()
    • get_optimizer()
    • invert_normalize_data()
    • load_augment_tensors()
    • load_model()
    • load_tensors()
    • model_init()
    • normalize_data()
    • numpy_to_tensor()
    • save_loss_components()
    • save_model()
    • select_features()
    • train_val_split()
  • bead.src.utils.loss module
    • BaseLoss
      • BaseLoss.calculate()
    • BinaryCrossEntropyLoss
      • BinaryCrossEntropyLoss.calculate()
    • DVAEFlowLoss
    • DVAELoss
    • KLDivergenceLoss
      • KLDivergenceLoss.calculate()
      • KLDivergenceLoss.compute_alpha_laplace()
    • L1Regularization
      • L1Regularization.calculate()
    • L2Regularization
      • L2Regularization.calculate()
    • ReconstructionLoss
      • ReconstructionLoss.calculate()
    • SupervisedContrastiveLoss
      • SupervisedContrastiveLoss.calculate()
    • VAEFlowLoss
      • VAEFlowLoss.calculate()
    • VAEFlowLossEMD
      • VAEFlowLossEMD.calculate()
    • VAEFlowLossL1
      • VAEFlowLossL1.calculate()
    • VAEFlowLossL2
      • VAEFlowLossL2.calculate()
    • VAEFlowSupConLoss
      • VAEFlowSupConLoss.calculate()
    • VAELoss
      • VAELoss.calculate()
    • VAELossEMD
      • VAELossEMD.calculate()
    • VAELossL1
      • VAELossL1.calculate()
    • VAELossL2
      • VAELossL2.calculate()
    • VAESupConLoss
      • VAESupConLoss.calculate()
    • WassersteinLoss
      • WassersteinLoss.calculate()
  • bead.src.utils.normalization module
    • invert_normalize_constit_pj_custom()
    • invert_normalize_jet_pj_custom()
    • normalize_constit_pj_custom()
    • normalize_jet_pj_custom()
  • bead.src.utils.plotting module
    • plot_latent_variables()
    • plot_losses()
    • plot_mu_logvar()
    • plot_roc_curve()
    • reduce_dim_subsampled()
  • Module contents

Module contents