bead.src package
Subpackages
- bead.src.models package
- Submodules
- bead.src.models.flows module
- bead.src.models.layers module
- bead.src.models.models module
AEAE_Dropout_BNConvAEConvFlow_ConvVAEConvVAEDirichlet_ConvVAEFlexibleTransformerHouseholderSylvester_ConvVAEIAF_ConvVAENSFAR_ConvVAEOrthogonalSylvester_ConvVAEPlanar_ConvVAETransformerAETriangularSylvester_ConvVAEVAEWithTransformercreate_flexible_transformer_from_config()create_transformer_config_from_config()create_vae_with_transformer_from_config()get_activation_function()get_output_activation_function()
- Module contents
- bead.src.trainers package
- bead.src.utils package
- Submodules
- bead.src.utils.conversion module
- bead.src.utils.data_processing module
- bead.src.utils.diagnostics module
- bead.src.utils.ggl module
ConfigConfig.activation_extractionConfig.batch_sizeConfig.chunk_sizeConfig.contrastive_temperatureConfig.contrastive_weightConfig.deterministic_algorithmConfig.early_stoppin_patienceConfig.early_stoppingConfig.efp_betaConfig.efp_beta_listConfig.efp_cache_dirConfig.efp_deviceConfig.efp_dmaxConfig.efp_dropout_rateConfig.efp_embedding_dimConfig.efp_epsConfig.efp_extended_modeConfig.efp_feature_prefixConfig.efp_gate_thresholdConfig.efp_gate_typeConfig.efp_include_compositesConfig.efp_measureConfig.efp_modeConfig.efp_monitor_sparsityConfig.efp_n_jobsConfig.efp_nmaxConfig.efp_normedConfig.efp_standardize_meanvarConfig.efp_use_layer_normConfig.efp_use_true_energyConfig.enable_transformerConfig.epochsConfig.file_typeConfig.input_featuresConfig.input_levelConfig.intermittent_model_savingConfig.intermittent_saving_patienceConfig.invert_normalizationsConfig.latent_space_plot_styleConfig.latent_space_sizeConfig.loss_functionConfig.lrConfig.lr_schedulerConfig.lr_scheduler_patienceConfig.min_deltaConfig.model_initConfig.model_nameConfig.normalizationsConfig.ntxent_sigmaConfig.num_constitsConfig.num_jetsConfig.optimizerConfig.overlay_rocConfig.overlay_roc_filenameConfig.overlay_roc_projectsConfig.overlay_roc_save_locationConfig.parallel_workersConfig.project_nameConfig.reg_paramConfig.separate_model_savingConfig.subsample_plotConfig.subsample_sizeConfig.train_sizeConfig.transformer_activationConfig.transformer_attention_dropoutConfig.transformer_d_ffConfig.transformer_d_modelConfig.transformer_dropoutConfig.transformer_gradient_checkpointingConfig.transformer_layer_norm_epsConfig.transformer_max_jetsConfig.transformer_modeConfig.transformer_n_headsConfig.transformer_n_layersConfig.transformer_norm_firstConfig.transformer_output_activationConfig.transformer_output_dimConfig.transformer_positional_encodingConfig.transformer_token_type_embeddingsConfig.transformer_use_class_attentionConfig.use_ampConfig.use_ddpConfig.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
ChainedScalerCustomDatasetEarlyStoppingL2NormalizerLRSchedulerLog1pScalerSinCosTransformeradd_sig_bkg_label()calculate_in_shape()call_forward()convert_to_tensor()create_datasets()data_label_split()detach_device()get_device()get_loss()get_ntxent_outputs()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()unpack_model_outputs()
- bead.src.utils.loss module
BaseLossBinaryCrossEntropyLossDVAEFlowLossDVAELossKLDivergenceLossL1RegularizationL2RegularizationNTXentCombinedLossNTXentDVAEFlowLossNTXentDVAELossNTXentLossNTXentVAEFlowLossNTXentVAELossReconstructionLossSupervisedContrastiveLossVAEFlowLossVAEFlowLossEMDVAEFlowLossL1VAEFlowLossL2VAEFlowSupConLossVAELossVAELossEMDVAELossL1VAELossL2VAESupConLossWassersteinLoss
- bead.src.utils.normalization module
- bead.src.utils.plotting module
- Module contents