Index

A

activations (umami.train_tools.configuration.NNStructureConfig attribute) add_compositor() (umami.tools.yaml_tools.CompositingComposer class method) add_data() (in module umami.preprocessing_tools.merging) add_eval_variables (umami.train_tools.configuration.EvaluationSettingsConfig attribute) add_filter() (umami.tools.yaml_tools.ExcludingConstructor class method) AddDataTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5) apply_atlas_style (umami.preprocessing_tools.configuration.GeneralSettings attribute) apply_scaling_jets() (in module umami.preprocessing_tools.scaling) apply_scaling_trks() (in module umami.preprocessing_tools.scaling) ApplyScalingJetsTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_scaling) ApplyScalingTrksTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_scaling) as_dict() (umami.preprocessing_tools.configuration.GeneralSettings method) (umami.preprocessing_tools.configuration.Sampling method) (umami.preprocessing_tools.configuration.SamplingOptions method)

as_full() (in module umami.preprocessing_tools.scaling) atlas_first_tag (umami.preprocessing_tools.configuration.GeneralSettings attribute) (umami.train_tools.configuration.ValidationSettingsConfig attribute) atlas_second_tag (umami.preprocessing_tools.configuration.GeneralSettings attribute) (umami.train_tools.configuration.ValidationSettingsConfig attribute) atoi() (in module umami.tools.tools) Attention (class in umami.tf_tools.layers) attention_condition (umami.train_tools.configuration.NNStructureConfig attribute) attention_sizes (umami.train_tools.configuration.NNStructureConfig attribute) AttentionPooling (class in umami.tf_tools.layers) AttentionTestCase (class in umami.tests.unit.tf_tools.test_tf_tools)

B

BaseGeneratorTest (class in umami.tests.unit.tf_tools.test_generators) batch_normalisation (umami.train_tools.configuration.NNStructureConfig attribute) batch_size (umami.train_tools.configuration.NNStructureConfig attribute) better_shuffling() (umami.preprocessing_tools.writing_train_file.TrainSampleWriter method)

binarise_jet_labels() (in module umami.preprocessing_tools.utils) BinariseJetLabelsTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools) bold_red (umami.configuration.configuration.CustomFormatter attribute) bool_attach_sample_weights (umami.preprocessing_tools.configuration.SamplingOptions attribute) build() (umami.tf_tools.layers.Sum method)

C

CadsGenerator (class in umami.tf_tools.generators) calc_disc_values() (in module umami.metrics.metrics) calc_validation_metrics() (in module umami.train_tools.nn_tools) CalcDiscValuesTestCase (class in umami.tests.unit.metrics.test_metrics) calculate_binning() (in module umami.preprocessing_tools.resampling.resampling_base) calculate_fraction_dict() (in module umami.evaluation_tools.eval_tools) calculate_pdf() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) calculate_pdf_ratio() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) calculate_saliency (umami.train_tools.configuration.EvaluationSettingsConfig attribute) calculate_weights() (umami.preprocessing_tools.writing_train_file.TrainSampleWriter method) (umami.tf_tools.tddgenerators.TDDGenerator method) CalculateBinningTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_resampling) CalculateScaling (class in umami.preprocessing_tools.scaling) call() (umami.tf_tools.layers.Attention method) (umami.tf_tools.layers.AttentionPooling method) (umami.tf_tools.layers.ConditionalAttention method) (umami.tf_tools.layers.ConditionalDeepSet method) (umami.tf_tools.layers.DeepSet method) (umami.tf_tools.layers.DenseNet method) (umami.tf_tools.layers.MaskedAverage1DPooling method) (umami.tf_tools.layers.MaskedSoftmax method) (umami.tf_tools.layers.Sum method) CallbackBase (class in umami.train_tools.nn_tools) CallbackBaseTestCase (class in umami.tests.unit.train_tools.test_nn_tools) category (umami.preprocessing_tools.configuration.Sample attribute) check_class_labels() (umami.train_tools.configuration.NNStructureConfig method) check_key() (in module umami.preprocessing_tools.configuration) check_keys() (in module umami.preprocessing_tools.merging) check_kwargs_for_ylabel_and_n_ratio_panel() (in module umami.input_vars_tools.plotting_functions) check_main_class_input() (in module umami.tools.tools) check_n_conditions() (umami.train_tools.configuration.NNStructureConfig method) check_option_definition() (in module umami.tools.tools) check_options() (umami.train_tools.configuration.NNStructureConfig method) check_resampling_options() (umami.preprocessing_tools.configuration.PreprocessConfiguration method) check_sample_consistency() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) check_shapes() (in module umami.preprocessing_tools.merging) check_size() (in module umami.preprocessing_tools.merging) check_train_file_format() (in module umami.train) CheckKeysTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5) CheckMainClassInputTestCase (class in umami.tests.unit.helper_tools.test_tools) CheckShapesTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5) CheckSizeTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5) class_labels (umami.preprocessing_tools.configuration.Sampling attribute) (umami.train_tools.configuration.NNStructureConfig attribute) combine_flavours() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) compare_h5_files_variables() (in module umami.data_tools.tools) compare_leading_spaces() (in module umami.tools.tools)

CompareH5FilesVariablesTestCase (class in umami.tests.unit.data_tools.test_data_tools_tools) compose() (umami.tools.yaml_tools.YAML method) compose_mapping_node() (umami.tools.yaml_tools.CompositingComposer method) compose_scalar_node() (umami.tools.yaml_tools.CompositingComposer method) compose_sequence_node() (umami.tools.yaml_tools.CompositingComposer method) CompositingComposer (class in umami.tools.yaml_tools) compositors (umami.tools.yaml_tools.CompositingComposer attribute) compression (umami.preprocessing_tools.configuration.GeneralSettings attribute) compute_mask() (umami.tf_tools.layers.Attention method) (umami.tf_tools.layers.ConditionalAttention method) (umami.tf_tools.layers.ConditionalDeepSet method) (umami.tf_tools.layers.DeepSet method) (umami.tf_tools.layers.MaskedSoftmax method) (umami.tf_tools.layers.Sum method) compute_output_shape() (umami.tf_tools.layers.Sum method) concat_jet_tracks (umami.preprocessing_tools.configuration.GeneralSettings attribute) concatenate_samples() (umami.preprocessing_tools.resampling.resampling_base.ResamplingTools method) ConditionalAttention (class in umami.tf_tools.layers) ConditionalDeepSet (class in umami.tf_tools.layers) ConditionalDeepSetTestCase (class in umami.tests.unit.tf_tools.test_tf_tools) config_file (umami.tests.unit.preprocessing.test_preprocessing.PreprocessingTestGetScaleDict attribute) (umami.tests.unit.preprocessing.test_preprocessing.PreprocessingTestParser attribute) config_path (umami.configuration.base.Configuration property) ConfigObject (class in umami.tests.unit.tf_tools.test_tf_tools) (class in umami.tests.unit.train_tools.test_nn_tools) Configuration (class in umami.configuration.base) ConfigurationTestCase (class in umami.tests.unit.train_tools.test_nn_tools) construct_mapping() (umami.tools.yaml_tools.ExcludingConstructor method) construct_sequence() (umami.tools.yaml_tools.ExcludingConstructor method) continue_training (umami.train_tools.configuration.TrainConfigurationObject attribute) convert_to_tfrecord (umami.preprocessing_tools.configuration.GeneralSettings attribute) ConvertTest (class in umami.tests.unit.tf_tools.test_tf_tools_convert_to_record) copy_to_out_dir() (umami.preprocessing_tools.configuration.PreprocessConfiguration method) correct_fractions() (in module umami.preprocessing_tools.resampling.resampling_base) CorrectFractionsTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_resampling) create_cads_model() (in module umami.models.model_cads) create_datasets() (in module umami.preprocessing_tools.merging) create_dips_model() (in module umami.models.model_dips) create_dl1_model() (in module umami.models.model_dl1) create_metadata_folder() (in module umami.train_tools.nn_tools) create_umami_cond_att_model() (in module umami.models.model_umami_cond_att) create_umami_model() (in module umami.models.model_umami) CreateDatasetsTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5) CreateMetadataFolderTestCase (class in umami.tests.unit.train_tools.test_nn_tools) CreateSamplesTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools) custom_n_jets_initial (umami.preprocessing_tools.configuration.SamplingOptions attribute) CustomFormatter (class in umami.configuration.configuration) cuts (umami.preprocessing_tools.configuration.Sample attribute)

D

date_format (umami.configuration.configuration.CustomFormatter attribute) debugformat (umami.configuration.configuration.CustomFormatter attribute) decode_fn() (umami.tf_tools.load_tfrecord.TFRecordReader method) DeepSet (class in umami.tf_tools.layers) deepsets_model() (in module umami.tf_tools.models) deepsets_model_umami() (in module umami.tf_tools.models) DeepSetTestCase (class in umami.tests.unit.tf_tools.test_tf_tools) default() (umami.preprocessing_tools.resampling.resampling_base.JsonNumpyEncoder method) dense_condition (umami.train_tools.configuration.NNStructureConfig attribute) dense_sizes (umami.train_tools.configuration.NNStructureConfig attribute) DenseNet (class in umami.tf_tools.layers) DenseNetTestCase (class in umami.tests.unit.tf_tools.test_tf_tools)

dict_file (umami.preprocessing_tools.configuration.GeneralSettings attribute) dips_dense_units (umami.train_tools.configuration.NNStructureConfig attribute) dips_loss_weight (umami.train_tools.configuration.NNStructureConfig attribute) dips_ppm_condition (umami.train_tools.configuration.NNStructureConfig attribute) dips_ppm_units (umami.train_tools.configuration.NNStructureConfig attribute) DipsGenerator (class in umami.tf_tools.generators) discriminant_output_shape() (in module umami.metrics.metrics) dl1_units (umami.train_tools.configuration.NNStructureConfig attribute) Dl1Generator (class in umami.tf_tools.generators) dropout_rate (umami.train_tools.configuration.NNStructureConfig attribute) dropout_rate_f (umami.train_tools.configuration.NNStructureConfig attribute) dropout_rate_phi (umami.train_tools.configuration.NNStructureConfig attribute)

E

eff_max (umami.train_tools.configuration.EvaluationSettingsConfig attribute) eff_min (umami.train_tools.configuration.EvaluationSettingsConfig attribute) epochs (umami.train_tools.configuration.NNStructureConfig attribute) eval_batch_size (umami.train_tools.configuration.EvaluationSettingsConfig attribute) EvalPlotsTestCase (class in umami.tests.unit.plotting_tools.test_eval_plot_functions) evaluate_model() (in module umami.evaluate_model) (in module umami.train_tools.nn_tools) evaluate_model_umami() (in module umami.train_tools.nn_tools) evaluate_traind_model (umami.train_tools.configuration.EvaluationSettingsConfig attribute)

evaluate_trained_model (umami.train_tools.configuration.NNStructureConfig attribute) (umami.train_tools.configuration.TrainConfigurationObject attribute) EvaluationSettingsConfig (class in umami.train_tools.configuration) event_indices() (in module umami.preprocessing_tools.ttbar_merge) event_list() (in module umami.preprocessing_tools.ttbar_merge) exclude (umami.train_tools.configuration.TrainConfigurationObject attribute) exclude_filter() (in module umami.tools.yaml_tools) ExcludingConstructor (class in umami.tools.yaml_tools) extra_classes_to_evaluate (umami.train_tools.configuration.EvaluationSettingsConfig attribute)

F

figsize (umami.train_tools.configuration.EvaluationSettingsConfig attribute) (umami.train_tools.configuration.ValidationSettingsConfig attribute) file_to_histogram() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) filter_dictionary() (in module umami.tf_tools.tddgenerators) filters (umami.tools.yaml_tools.ExcludingConstructor attribute) flatten() (in module umami.tools.tools) flatten_list() (in module umami.tools.tools) FlattenListTest (class in umami.tests.unit.tools.test_tools)

fork() (umami.tools.yaml_tools.YAML method) format() (umami.configuration.configuration.CustomFormatter method) FORMATS (umami.configuration.configuration.CustomFormatter attribute) frac_max (umami.train_tools.configuration.EvaluationSettingsConfig attribute) frac_min (umami.train_tools.configuration.EvaluationSettingsConfig attribute) frac_step (umami.train_tools.configuration.EvaluationSettingsConfig attribute) frac_values (umami.train_tools.configuration.EvaluationSettingsConfig attribute) frac_values_comp (umami.train_tools.configuration.EvaluationSettingsConfig attribute) fractions (umami.preprocessing_tools.configuration.SamplingOptions attribute)

G

GeneralSettings (class in umami.preprocessing_tools.configuration) generate_flavour_pdf() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) generate_number_sample() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) generate_target_pdf() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) generate_test_data() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.ResamplingTestCaseComplexDistr method) (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.ResamplingTestCaseEasyDistr method) get_batches_per_file() (umami.preprocessing_tools.preparation.PrepareSamples method) get_bins() (umami.preprocessing_tools.resampling.resampling_base.Resampling method) get_category_cuts() (in module umami.data_tools.cuts) get_class_label_variables() (in module umami.helper_tools.classification_tools) get_class_prob_var_names() (in module umami.helper_tools.classification_tools) get_comp_tagger_rej_dict() (in module umami.plotting_tools.train_plotting_functions) get_compositor() (umami.tools.yaml_tools.CompositingComposer class method) get_config() (umami.tf_tools.layers.Attention method) (umami.tf_tools.layers.ConditionalAttention method) (umami.tf_tools.layers.ConditionalDeepSet method) (umami.tf_tools.layers.DeepSet method) (umami.tf_tools.layers.DenseNet method) (umami.tf_tools.layers.MaskedSoftmax method) get_configuration() (in module umami.tests.integration.test_input_vars_plot) (in module umami.tests.integration.test_plotting_umami) (in module umami.tests.integration.test_preprocessing) (in module umami.tests.integration.test_preprocessing_upp) (in module umami.tests.integration.test_train) (umami.configuration.configuration.GlobalConfiguration method) (umami.preprocessing_tools.configuration.PreprocessConfiguration method) (umami.train_tools.configuration.TrainConfiguration method) get_cut_list() (in module umami.data_tools.cuts) get_datasets_configuration() (in module umami.input_vars_tools.plotting_functions) get_dropout_rates() (in module umami.train_tools.nn_tools) get_epoch_from_string() (in module umami.train_tools.nn_tools) get_file_name() (umami.preprocessing_tools.configuration.PreprocessConfiguration method) get_flavour_weights() (umami.preprocessing_tools.resampling.weighting.Weighting method) get_generator() (in module umami.tf_tools.tddgenerators) get_gradients() (in module umami.metrics.metrics) get_indices() (umami.preprocessing_tools.resampling.count_sampling.SimpleSamplingBase method) (umami.preprocessing_tools.resampling.count_sampling.UnderSampling method) (umami.preprocessing_tools.resampling.importance_sampling_no_replace.UnderSamplingNoReplace method) (umami.preprocessing_tools.resampling.weighting.Weighting method) (umami.preprocessing_tools.ttbar_merge.TTbarMerge method) get_input_files() (umami.preprocessing_tools.configuration.Preparation method) (umami.preprocessing_tools.ttbar_merge.TTbarMerge method) get_jet_feature_indices() (in module umami.train_tools.nn_tools) get_jet_feature_position() (in module umami.train_tools.nn_tools) get_learning_rate_reducer() (in module umami.tf_tools.tools) get_metrics_file_name() (in module umami.train_tools.nn_tools) get_model_path() (in module umami.train_tools.nn_tools) get_n_dim() (umami.tf_tools.generators.ModelGenerator method) (umami.tf_tools.tddgenerators.TDDGenerator method) get_n_jet_features() (umami.tf_tools.generators.ModelGenerator method) (umami.tf_tools.tddgenerators.TDDGenerator method) get_n_jets() (umami.tf_tools.generators.ModelGenerator method) (umami.tf_tools.tddgenerators.TDDGenerator method) get_n_trk_features() (umami.tf_tools.generators.ModelGenerator method) (umami.tf_tools.tddgenerators.TDDGenerator method) get_n_trks() (umami.tf_tools.generators.ModelGenerator method) (umami.tf_tools.tddgenerators.TDDGenerator method)

get_normalisation_arrays() (umami.tf_tools.tddgenerators.TDDGenerator method) get_parameters_from_validation_dict_name() (in module umami.train_tools.nn_tools) get_parser() (in module umami.evaluate_model) (in module umami.plot_input_variables) (in module umami.plotting_epoch_performance) (in module umami.plotting_umami) (in module umami.preprocessing) (in module umami.sample_merging) (in module umami.train) get_pt_eta_bin_statistics() (umami.preprocessing_tools.resampling.resampling_base.ResamplingTools method) get_rej_per_eff_dict() (in module umami.evaluation_tools.eval_tools) get_rej_per_frac_dict() (in module umami.evaluation_tools.eval_tools) get_rejection() (in module umami.metrics.metrics) get_saliency_map_dict() (in module umami.evaluation_tools.eval_tools) get_sample() (umami.preprocessing_tools.configuration.Preparation method) get_sample_cuts() (in module umami.data_tools.cuts) get_sampling_probabilities() (umami.preprocessing_tools.resampling.importance_sampling_no_replace.UnderSamplingNoReplace method) get_sampling_probability() (umami.preprocessing_tools.resampling.importance_sampling_no_replace.UnderSamplingNoReplace method) get_scale_dict() (in module umami.preprocessing_tools.utils) (umami.preprocessing_tools.scaling.CalculateScaling method) get_scaling() (umami.preprocessing_tools.scaling.CalculateScaling method) get_scaling_dict_generator() (umami.preprocessing_tools.scaling.CalculateScaling method) get_scaling_tracks() (umami.preprocessing_tools.scaling.CalculateScaling method) get_scaling_tracks_generator() (umami.preprocessing_tools.scaling.CalculateScaling method) get_score() (in module umami.metrics.metrics) get_scores_probs_dict() (in module umami.evaluation_tools.eval_tools) get_size() (in module umami.preprocessing_tools.merging) get_test_file() (in module umami.train_tools.nn_tools) get_test_sample() (in module umami.train_tools.nn_tools) get_test_sample_trks() (in module umami.train_tools.nn_tools) get_track_mask() (in module umami.preprocessing_tools.scaling) get_track_vars() (umami.tf_tools.tddgenerators.TDDGenerator method) get_unique_identifiers() (in module umami.train_tools.nn_tools) get_valid_class_categories() (umami.preprocessing_tools.resampling.resampling_base.ResamplingTools method) get_variable_dict() (in module umami.preprocessing_tools.utils) GetCategoryCutsTestCase (class in umami.tests.unit.data_tools.test_data_tools_cuts) GetClassTestCase (class in umami.tests.unit.helper_tools.test_classification_tools) GetCompTaggerRejDictTestCase (class in umami.tests.unit.plotting_tools.test_train_plot_functions) GetDictTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools) GetDropoutRatesTestCase (class in umami.tests.unit.train_tools.test_nn_tools) GetEpochFromStringTestCase (class in umami.tests.unit.train_tools.test_nn_tools) GetJetFeatureIndicesTestCase (class in umami.tests.unit.train_tools.test_nn_tools) GetJetFeaturePositionTestCase (class in umami.tests.unit.train_tools.test_nn_tools) GetMetricsFileNameTestCase (class in umami.tests.unit.train_tools.test_nn_tools) GetModelPathTestCase (class in umami.tests.unit.train_tools.test_nn_tools) GetRejectionTestCase (class in umami.tests.unit.metrics.test_metrics) GetSampleCutsTestCase (class in umami.tests.unit.data_tools.test_data_tools_cuts) GetSamplesTestCase (class in umami.tests.unit.train_tools.test_nn_tools) GetScoreTestCase (class in umami.tests.unit.metrics.test_metrics) GetSizeTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5) GetUniqueIdentifiersTestCase (class in umami.tests.unit.train_tools.test_nn_tools) GetVariableDictTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools) GlobalConfiguration (class in umami.configuration.configuration) green (umami.configuration.configuration.CustomFormatter attribute) grey (umami.configuration.configuration.CustomFormatter attribute)

H

H5ToTFRecords (class in umami.tf_tools.convert_to_record) HelperFunctionTestCase (class in umami.tests.unit.input_vars_tools.test_input_vars_tools) hist_ratio() (in module umami.helper_tools.histogram_tools)

hist_w_unc() (in module umami.helper_tools.histogram_tools) HistRatioTestCase (class in umami.tests.unit.helper_tools.test_histogram_tools) HistWUncTestCase (class in umami.tests.unit.helper_tools.test_histogram_tools)

I

in_memory_resample() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) include_compositor() (in module umami.tools.yaml_tools) init_jet_datasets() (umami.preprocessing_tools.writing_train_file.TrainSampleWriter method) init_track_datasets() (umami.preprocessing_tools.writing_train_file.TrainSampleWriter method) initialise_flavour_samples() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method)

initialise_samples() (umami.preprocessing_tools.resampling.resampling_base.ResamplingTools method) inter_func (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling property) intermediate_index_file (umami.preprocessing_tools.configuration.SamplingOptions attribute) intermediate_index_file_validation (umami.preprocessing_tools.configuration.SamplingOptions attribute) intermediate_units (umami.train_tools.configuration.NNStructureConfig attribute)

J

JetPlottingTestCase (class in umami.tests.unit.input_vars_tools.test_input_vars_tools) jets_generator() (umami.preprocessing_tools.preparation.PrepareSamples method) join() (in module umami.tools.yaml_tools)

join_scale_dicts() (umami.preprocessing_tools.scaling.CalculateScaling method) join_structured_arrays() (in module umami.preprocessing_tools.utils) JsonNumpyEncoder (class in umami.preprocessing_tools.resampling.resampling_base)

L

learning_rate (umami.train_tools.configuration.NNStructureConfig attribute) legend_sample_category (umami.preprocessing_tools.configuration.GeneralSettings attribute) load() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) load_config_file() (umami.configuration.base.Configuration method) (umami.configuration.configuration.GlobalConfiguration method) (umami.train_tools.configuration.TrainConfiguration method) load_dataset() (umami.tf_tools.load_tfrecord.TFRecordReader method) load_file() (umami.tests.unit.helper_tools.test_tools.ReplaceLineInFileTestCase method) load_h5file_train() (umami.tf_tools.convert_to_record.H5ToTFRecords method) load_in_memory() (umami.tf_tools.generators.ModelGenerator method) (umami.tf_tools.tddgenerators.TDDGenerator method) load_index_generator() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) load_jets_from_file() (in module umami.data_tools.loaders) load_jets_generator() (umami.preprocessing_tools.ttbar_merge.TTbarMerge method) load_optimiser (umami.train_tools.configuration.NNStructureConfig attribute)

load_samples() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) load_samples_generator() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) load_scaled_generator() (umami.preprocessing_tools.writing_train_file.TrainSampleWriter method) load_tfrecords_train_dataset() (in module umami.tf_tools.load_tfrecord) load_trks_from_file() (in module umami.data_tools.loaders) load_validation_data() (in module umami.train_tools.nn_tools) LoadFilesTestCase (class in umami.tests.unit.data_tools.test_data_tools_loaders) lrr (umami.train_tools.configuration.NNStructureConfig attribute) lrr_cooldown (umami.train_tools.configuration.NNStructureConfig attribute) lrr_factor (umami.train_tools.configuration.NNStructureConfig attribute) lrr_min_learning_rate (umami.train_tools.configuration.NNStructureConfig attribute) lrr_mode (umami.train_tools.configuration.NNStructureConfig attribute) lrr_monitor (umami.train_tools.configuration.NNStructureConfig attribute) lrr_patience (umami.train_tools.configuration.NNStructureConfig attribute) lrr_verbose (umami.train_tools.configuration.NNStructureConfig attribute)

M

main() (in module umami.plotting_epoch_performance) main_class (umami.train_tools.configuration.NNStructureConfig attribute) MaskedAverage1DPooling (class in umami.tf_tools.layers) MaskedSoftmax (class in umami.tf_tools.layers) max_upsampling_ratio (umami.preprocessing_tools.configuration.SamplingOptions attribute) merge() (umami.preprocessing_tools.ttbar_merge.TTbarMerge method) MergeConfig (class in umami.preprocessing_tools.ttbar_merge) method (umami.preprocessing_tools.configuration.Sampling attribute) MetricsSmallTestCase (class in umami.tests.unit.metrics.test_metrics) MinimalPreprocessingConfig (class in umami.tests.unit.train_tools.test_nn_tools) model_file (umami.train_tools.configuration.TrainConfigurationObject attribute) model_name (umami.train_tools.configuration.TrainConfigurationObject attribute) ModelGenerator (class in umami.tf_tools.generators) module umami umami.configuration umami.configuration.base umami.configuration.configuration umami.data_tools umami.data_tools.cuts umami.data_tools.loaders umami.data_tools.tools umami.evaluate_model umami.evaluation_tools umami.evaluation_tools.eval_tools umami.evaluation_tools.feature_importance umami.helper_tools umami.helper_tools.classification_tools umami.helper_tools.histogram_tools umami.input_vars_tools umami.input_vars_tools.plotting_functions umami.metrics umami.metrics.metrics umami.models umami.models.model_cads umami.models.model_dips umami.models.model_dl1 umami.models.model_umami umami.models.model_umami_cond_att umami.plot_input_variables umami.plotting_epoch_performance umami.plotting_tools umami.plotting_tools.eval_plotting_functions umami.plotting_tools.preprocessing_plotting_functions umami.plotting_tools.train_plotting_functions umami.plotting_tools.utils umami.plotting_umami umami.preprocessing umami.preprocessing_tools umami.preprocessing_tools.configuration umami.preprocessing_tools.merging umami.preprocessing_tools.preparation umami.preprocessing_tools.resampling umami.preprocessing_tools.resampling.count_sampling umami.preprocessing_tools.resampling.importance_sampling_no_replace umami.preprocessing_tools.resampling.pdf_sampling umami.preprocessing_tools.resampling.resampling_base umami.preprocessing_tools.resampling.weighting umami.preprocessing_tools.scaling umami.preprocessing_tools.ttbar_merge umami.preprocessing_tools.utils umami.preprocessing_tools.writing_train_file umami.sample_merging umami.tests umami.tests.integration umami.tests.integration.test_examples umami.tests.integration.test_input_vars_plot umami.tests.integration.test_plotting_umami umami.tests.integration.test_preprocessing umami.tests.integration.test_preprocessing_upp umami.tests.integration.test_train umami.tests.unit umami.tests.unit.data_tools umami.tests.unit.data_tools.test_data_tools_cuts umami.tests.unit.data_tools.test_data_tools_loaders umami.tests.unit.data_tools.test_data_tools_tools umami.tests.unit.evaluation_tools umami.tests.unit.evaluation_tools.test_eval_tools umami.tests.unit.helper_tools umami.tests.unit.helper_tools.test_classification_tools umami.tests.unit.helper_tools.test_histogram_tools umami.tests.unit.helper_tools.test_tools umami.tests.unit.input_vars_tools umami.tests.unit.input_vars_tools.test_input_vars_tools umami.tests.unit.metrics umami.tests.unit.metrics.test_metrics umami.tests.unit.plotting_tools umami.tests.unit.plotting_tools.test_eval_plot_functions umami.tests.unit.plotting_tools.test_plotting_utils umami.tests.unit.plotting_tools.test_preprocessing_plotting_functions umami.tests.unit.plotting_tools.test_train_plot_functions umami.tests.unit.preprocessing umami.tests.unit.preprocessing.test_preprocessing umami.tests.unit.preprocessing.test_preprocessing_tools umami.tests.unit.preprocessing.test_preprocessing_tools_configuration umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5 umami.tests.unit.preprocessing.test_preprocessing_tools_resampling umami.tests.unit.preprocessing.test_preprocessing_tools_scaling umami.tests.unit.tf_tools umami.tests.unit.tf_tools.test_generators umami.tests.unit.tf_tools.test_tddgenerators umami.tests.unit.tf_tools.test_tf_tools umami.tests.unit.tf_tools.test_tf_tools_convert_to_record umami.tests.unit.tools umami.tests.unit.tools.test_tools umami.tests.unit.train_tools umami.tests.unit.train_tools.test_nn_tools umami.tf_tools umami.tf_tools.convert_to_record umami.tf_tools.generators umami.tf_tools.layers umami.tf_tools.load_tfrecord umami.tf_tools.models umami.tf_tools.tddgenerators umami.tf_tools.tools umami.tools umami.tools.tools umami.tools.yaml_tools umami.train umami.train_tools umami.train_tools.configuration umami.train_tools.nn_tools

MyCallback (class in umami.train_tools.nn_tools) MyCallbackTestCase (class in umami.tests.unit.train_tools.test_nn_tools) MyCallbackUmami (class in umami.train_tools.nn_tools) MyCallbackUmamiTestCase (class in umami.tests.unit.train_tools.test_nn_tools)

N

n_conditions (umami.train_tools.configuration.NNStructureConfig attribute) n_jets (umami.preprocessing_tools.configuration.Sample attribute) (umami.preprocessing_tools.configuration.SamplingOptions attribute) (umami.train_tools.configuration.EvaluationSettingsConfig attribute) (umami.train_tools.configuration.ValidationSettingsConfig attribute) n_jets_scaling (umami.preprocessing_tools.configuration.SamplingOptions attribute)

n_jets_to_plot (umami.preprocessing_tools.configuration.SamplingOptions attribute) n_jets_train (umami.train_tools.configuration.NNStructureConfig attribute) n_jets_validation (umami.preprocessing_tools.configuration.SamplingOptions attribute) name (umami.preprocessing_tools.configuration.Sample attribute) natural_keys() (in module umami.tools.tools) nfiles_tfrecord (umami.train_tools.configuration.NNStructureConfig attribute) NNStructureConfig (class in umami.train_tools.configuration)

O

on_epoch_end() (umami.train_tools.nn_tools.MyCallback method) (umami.train_tools.nn_tools.MyCallbackUmami method) options (umami.preprocessing_tools.configuration.Sampling attribute)

outfile_name (umami.preprocessing_tools.configuration.GeneralSettings attribute) outfile_name_validation (umami.preprocessing_tools.configuration.GeneralSettings attribute) output_name (umami.preprocessing_tools.configuration.Sample attribute)

P

PDFResamplingTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_resampling) PDFSampling (class in umami.preprocessing_tools.resampling.pdf_sampling) plot_accuracies() (in module umami.plotting_tools.train_plotting_functions) plot_accuracies_umami() (in module umami.plotting_tools.train_plotting_functions) plot_args (umami.train_tools.configuration.ValidationSettingsConfig property) plot_confusion_matrix() (in module umami.plotting_tools.eval_plotting_functions) (in module umami.plotting_umami) plot_datatype (umami.train_tools.configuration.ValidationSettingsConfig attribute) plot_disc_cut_per_epoch() (in module umami.plotting_tools.train_plotting_functions) plot_disc_cut_per_epoch_umami() (in module umami.plotting_tools.train_plotting_functions) plot_frac_contour() (in module umami.plotting_umami) plot_fraction_contour() (in module umami.plotting_tools.eval_plotting_functions) plot_input_vars_jets() (in module umami.input_vars_tools.plotting_functions) plot_input_vars_trks() (in module umami.input_vars_tools.plotting_functions) plot_jets_variables() (in module umami.plot_input_variables) plot_losses() (in module umami.plotting_tools.train_plotting_functions) plot_losses_umami() (in module umami.plotting_tools.train_plotting_functions) plot_n_tracks_per_jet() (in module umami.input_vars_tools.plotting_functions) plot_name (umami.preprocessing_tools.configuration.GeneralSettings attribute) plot_options_as_dict() (umami.preprocessing_tools.configuration.GeneralSettings method) plot_prob() (in module umami.plotting_tools.eval_plotting_functions) plot_probability() (in module umami.plotting_umami) plot_rej_per_epoch() (in module umami.plotting_tools.train_plotting_functions) plot_rej_per_epoch_comp() (in module umami.plotting_tools.train_plotting_functions) plot_resampling_variables() (in module umami.plotting_tools.preprocessing_plotting_functions) plot_roc() (in module umami.plotting_tools.eval_plotting_functions) (in module umami.plotting_umami) plot_saliency() (in module umami.plotting_tools.eval_plotting_functions) (in module umami.plotting_umami)

plot_score() (in module umami.plotting_tools.eval_plotting_functions) (in module umami.plotting_umami) plot_trks_variables() (in module umami.plot_input_variables) plot_type (umami.preprocessing_tools.configuration.GeneralSettings attribute) plot_unique_jet_appearence() (in module umami.plotting_tools.preprocessing_plotting_functions) plot_validation_files() (in module umami.plotting_tools.train_plotting_functions) plot_var_vs_eff() (in module umami.plotting_tools.eval_plotting_functions) (in module umami.plotting_umami) plot_variable() (in module umami.plotting_tools.preprocessing_plotting_functions) PlotResamplingVariablesTestCase (class in umami.tests.unit.plotting_tools.test_preprocessing_plotting_functions) pooling (umami.train_tools.configuration.NNStructureConfig attribute) ppm_condition (umami.train_tools.configuration.NNStructureConfig attribute) ppm_sizes (umami.train_tools.configuration.NNStructureConfig attribute) precision (umami.preprocessing_tools.configuration.GeneralSettings attribute) Preparation (class in umami.preprocessing_tools.configuration) PreparationTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_configuration) prepare_config() (in module umami.tests.integration.test_train) prepare_model() (in module umami.tf_tools.models) PrepareModelTestCase (class in umami.tests.unit.tf_tools.test_tf_tools) PrepareSamples (class in umami.preprocessing_tools.preparation) PrepareSamplesTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools) PrepareSamplesTestCase.CArgs (class in umami.tests.unit.preprocessing.test_preprocessing_tools) preprocess_config (umami.train_tools.configuration.TrainConfigurationObject attribute) PreprocessConfiguration (class in umami.preprocessing_tools.configuration) PreprocessConfigurationTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_configuration) preprocessing_plots() (in module umami.plotting_tools.preprocessing_plotting_functions) PreprocessingPlotsTestCase (class in umami.tests.unit.plotting_tools.test_preprocessing_plotting_functions) PreprocessingTestGetScaleDict (class in umami.tests.unit.preprocessing.test_preprocessing) PreprocessingTestParser (class in umami.tests.unit.preprocessing.test_preprocessing)

Q

quick_check_duplicates() (in module umami.preprocessing_tools.resampling.resampling_base)

R

ratio (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling property) read_dataframe_repetition() (in module umami.preprocessing_tools.resampling.resampling_base) recompute_score() (in module umami.evaluation_tools.eval_tools) red (umami.configuration.configuration.CustomFormatter attribute) repeat_end (umami.train_tools.configuration.NNStructureConfig attribute) replace_line_in_file() (in module umami.tools.tools) ReplaceLineInFileTestCase (class in umami.tests.unit.helper_tools.test_tools) resample_chunk() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) resample_iterator() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) Resampling (class in umami.preprocessing_tools.resampling.resampling_base) ResamplingTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_resampling) ResamplingTestCaseComplexDistr (class in umami.tests.unit.preprocessing.test_preprocessing_tools_resampling) ResamplingTestCaseEasyDistr (class in umami.tests.unit.preprocessing.test_preprocessing_tools_resampling) ResamplingTools (class in umami.preprocessing_tools.resampling.resampling_base) reset (umami.configuration.configuration.CustomFormatter attribute)

results_filename_extension (umami.train_tools.configuration.EvaluationSettingsConfig attribute) retrieve_common_variables() (umami.preprocessing_tools.resampling.resampling_base.Resampling method) retrieve_cut_string() (in module umami.data_tools.cuts) RetrieveTruthLabelVarValueTestCase (class in umami.tests.unit.plotting_tools.test_plotting_utils) return_unnormalised_pdf_weights() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) run() (umami.preprocessing_tools.preparation.PrepareSamples method) Run() (umami.preprocessing_tools.resampling.count_sampling.SimpleSamplingBase method) (umami.preprocessing_tools.resampling.count_sampling.UnderSampling method) (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) (umami.preprocessing_tools.resampling.weighting.Weighting method) run_plot_input_vars() (in module umami.tests.integration.test_input_vars_plot) run_plotting() (in module umami.tests.integration.test_plotting_umami) run_preprocessing() (in module umami.tests.integration.test_preprocessing) (in module umami.tests.integration.test_preprocessing_upp) run_training() (in module umami.tests.integration.test_train) run_validation_check() (in module umami.plotting_tools.train_plotting_functions)

S

saliency_effs (umami.train_tools.configuration.EvaluationSettingsConfig attribute) saliency_ntrks (umami.train_tools.configuration.EvaluationSettingsConfig attribute) Sample (class in umami.preprocessing_tools.configuration) sample_flavour() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) samples_training (umami.preprocessing_tools.configuration.SamplingOptions attribute) samples_validation (umami.preprocessing_tools.configuration.SamplingOptions attribute) Sampling (class in umami.preprocessing_tools.configuration) sampling_fraction (umami.preprocessing_tools.configuration.SamplingOptions attribute) sampling_generator() (umami.preprocessing_tools.resampling.resampling_base.Resampling static method) sampling_variables (umami.preprocessing_tools.configuration.SamplingOptions attribute) SamplingGeneratorTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_resampling) SamplingOptions (class in umami.preprocessing_tools.configuration) save() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) save_chunk() (umami.preprocessing_tools.writing_train_file.TrainSampleWriter method) save_complete_iterator() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) save_divide() (in module umami.helper_tools.histogram_tools) save_flavour() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) save_parameters() (umami.tf_tools.convert_to_record.H5ToTFRecords method) save_partial_iterator() (umami.preprocessing_tools.resampling.pdf_sampling.PDFSampling method) save_track_labels (umami.preprocessing_tools.configuration.SamplingOptions attribute) save_tracks (umami.preprocessing_tools.configuration.SamplingOptions attribute) SaveDivideTestCase (class in umami.tests.unit.helper_tools.test_histogram_tools) scale_input() (umami.tf_tools.tddgenerators.TDDGenerator method) scale_tracks() (umami.tf_tools.tddgenerators.TDDGenerator method) ScalingTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_scaling) set_log_level() (in module umami.configuration.configuration) set_logging_level() (umami.configuration.configuration.GlobalConfiguration method) set_mpl_plotting_backend() (umami.configuration.configuration.GlobalConfiguration method) set_tf_debug_level() (umami.configuration.configuration.GlobalConfiguration method) set_up_plots() (in module umami.plotting_umami) setUp() (umami.tests.integration.test_input_vars_plot.TestInputVarsPlotting method) (umami.tests.integration.test_plotting_umami.TestPlottingUmami method) (umami.tests.integration.test_preprocessing.TestPreprocessing method) (umami.tests.integration.test_preprocessing_upp.TestPreprocessingUPP method) (umami.tests.integration.test_train.TestTraining method) (umami.tests.unit.data_tools.test_data_tools_cuts.GetCategoryCutsTestCase method) (umami.tests.unit.data_tools.test_data_tools_cuts.GetSampleCutsTestCase method) (umami.tests.unit.data_tools.test_data_tools_loaders.LoadFilesTestCase method) (umami.tests.unit.data_tools.test_data_tools_tools.CompareH5FilesVariablesTestCase method) (umami.tests.unit.evaluation_tools.test_eval_tools.SmallFunctionsTestCase method) (umami.tests.unit.helper_tools.test_classification_tools.GetClassTestCase method) (umami.tests.unit.helper_tools.test_histogram_tools.HistRatioTestCase method) (umami.tests.unit.helper_tools.test_histogram_tools.HistWUncTestCase method) (umami.tests.unit.helper_tools.test_tools.CheckMainClassInputTestCase method) (umami.tests.unit.helper_tools.test_tools.ReplaceLineInFileTestCase method) (umami.tests.unit.input_vars_tools.test_input_vars_tools.HelperFunctionTestCase method) (umami.tests.unit.input_vars_tools.test_input_vars_tools.JetPlottingTestCase method) (umami.tests.unit.metrics.test_metrics.CalcDiscValuesTestCase method) (umami.tests.unit.metrics.test_metrics.GetRejectionTestCase method) (umami.tests.unit.metrics.test_metrics.GetScoreTestCase method) (umami.tests.unit.metrics.test_metrics.MetricsSmallTestCase method) (umami.tests.unit.plotting_tools.test_eval_plot_functions.EvalPlotsTestCase method) (umami.tests.unit.plotting_tools.test_plotting_utils.RetrieveTruthLabelVarValueTestCase method) (umami.tests.unit.plotting_tools.test_preprocessing_plotting_functions.PlotResamplingVariablesTestCase method) (umami.tests.unit.plotting_tools.test_preprocessing_plotting_functions.PreprocessingPlotsTestCase method) (umami.tests.unit.plotting_tools.test_train_plot_functions.GetCompTaggerRejDictTestCase method) (umami.tests.unit.plotting_tools.test_train_plot_functions.TrainPlotsTestCase method) (umami.tests.unit.preprocessing.test_preprocessing.PreprocessingTestGetScaleDict method) (umami.tests.unit.preprocessing.test_preprocessing.PreprocessingTestParser method) (umami.tests.unit.preprocessing.test_preprocessing_tools.BinariseJetLabelsTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools.CreateSamplesTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools.GetDictTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools.GetVariableDictTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools.PrepareSamplesTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_configuration.PreparationTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_configuration.PreprocessConfigurationTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.AddDataTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.CheckKeysTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.CheckShapesTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.CheckSizeTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.CreateDatasetsTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.GetSizeTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.PDFResamplingTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.ResamplingTestCaseComplexDistr method) (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.ResamplingTestCaseEasyDistr method) (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.SamplingGeneratorTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.UnderSamplingNoReplaceTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_scaling.ApplyScalingJetsTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_scaling.ApplyScalingTrksTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_scaling.ScalingTestCase method) (umami.tests.unit.tf_tools.test_generators.BaseGeneratorTest method) (umami.tests.unit.tf_tools.test_tddgenerators.TestTDDGenerator method) (umami.tests.unit.tf_tools.test_tf_tools.AttentionTestCase method) (umami.tests.unit.tf_tools.test_tf_tools.ConditionalDeepSetTestCase method) (umami.tests.unit.tf_tools.test_tf_tools.DeepSetTestCase method) (umami.tests.unit.tf_tools.test_tf_tools.DenseNetTestCase method) (umami.tests.unit.tf_tools.test_tf_tools.PrepareModelTestCase method) (umami.tests.unit.tf_tools.test_tf_tools_convert_to_record.ConvertTest method) (umami.tests.unit.train_tools.test_nn_tools.CallbackBaseTestCase method) (umami.tests.unit.train_tools.test_nn_tools.ConfigurationTestCase method) (umami.tests.unit.train_tools.test_nn_tools.CreateMetadataFolderTestCase method) (umami.tests.unit.train_tools.test_nn_tools.GetDropoutRatesTestCase method) (umami.tests.unit.train_tools.test_nn_tools.GetEpochFromStringTestCase method) (umami.tests.unit.train_tools.test_nn_tools.GetJetFeatureIndicesTestCase method) (umami.tests.unit.train_tools.test_nn_tools.GetJetFeaturePositionTestCase method) (umami.tests.unit.train_tools.test_nn_tools.GetMetricsFileNameTestCase method) (umami.tests.unit.train_tools.test_nn_tools.GetModelPathTestCase method) (umami.tests.unit.train_tools.test_nn_tools.GetSamplesTestCase method) (umami.tests.unit.train_tools.test_nn_tools.GetUniqueIdentifiersTestCase method) (umami.tests.unit.train_tools.test_nn_tools.MyCallbackTestCase method) (umami.tests.unit.train_tools.test_nn_tools.MyCallbackUmamiTestCase method) (umami.tests.unit.train_tools.test_nn_tools.SetupOutputDirectoryTestCase method)

setup_output_directory() (in module umami.train_tools.nn_tools) setUpClass() (umami.tests.unit.preprocessing.test_preprocessing_tools.TTbarMergeTestCase class method) SetupOutputDirectoryTestCase (class in umami.tests.unit.train_tools.test_nn_tools) shapley (umami.train_tools.configuration.EvaluationSettingsConfig attribute) shapley_all_flavours() (in module umami.evaluation_tools.feature_importance) shapley_one_flavour() (in module umami.evaluation_tools.feature_importance) SimpleSamplingBase (class in umami.preprocessing_tools.resampling.count_sampling) SmallFunctionsTestCase (class in umami.tests.unit.evaluation_tools.test_eval_tools) Sum (class in umami.tf_tools.layers)

T

tagger (umami.train_tools.configuration.EvaluationSettingsConfig attribute) (umami.train_tools.configuration.NNStructureConfig attribute) tagger_label (umami.train_tools.configuration.ValidationSettingsConfig attribute) taggers_from_file (umami.train_tools.configuration.ValidationSettingsConfig attribute) target_distribution (umami.preprocessing_tools.configuration.SamplingOptions attribute) TDDCadsGenerator (class in umami.tf_tools.tddgenerators) TDDDipsGenerator (class in umami.tf_tools.tddgenerators) TDDDl1Generator (class in umami.tf_tools.tddgenerators) TDDGenerator (class in umami.tf_tools.tddgenerators) TDDUmamiConditionGenerator (class in umami.tf_tools.tddgenerators) TDDUmamiGenerator (class in umami.tf_tools.tddgenerators) tearDown() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.ResamplingTestCaseComplexDistr method) (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.ResamplingTestCaseEasyDistr method) (umami.tests.unit.tf_tools.test_tddgenerators.TestTDDGenerator method) tearDownClass() (umami.tests.unit.preprocessing.test_preprocessing_tools.TTbarMergeTestCase class method) test_2_classes_correct_output() (umami.tests.unit.preprocessing.test_preprocessing_tools.BinariseJetLabelsTestCase method) test_3_classes() (umami.tests.unit.plotting_tools.test_plotting_utils.RetrieveTruthLabelVarValueTestCase method) test_4_classes() (umami.tests.unit.plotting_tools.test_plotting_utils.RetrieveTruthLabelVarValueTestCase method) test_add_data() (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.AddDataTestCase method) test_already_flattened() (umami.tests.unit.tools.test_tools.FlattenListTest method) test_apply_scaling_trks() (umami.tests.unit.preprocessing.test_preprocessing_tools_scaling.ApplyScalingTrksTestCase method) test_cads_config() (umami.tests.unit.train_tools.test_nn_tools.ConfigurationTestCase method) test_cads_without_cond_info() (umami.tests.unit.train_tools.test_nn_tools.ConfigurationTestCase method) test_calc_disc_values() (umami.tests.unit.metrics.test_metrics.CalcDiscValuesTestCase method) test_calc_disc_values_2signal_classes_bkg_jets() (umami.tests.unit.metrics.test_metrics.CalcDiscValuesTestCase method) test_calc_disc_values_2singal_classes() (umami.tests.unit.metrics.test_metrics.CalcDiscValuesTestCase method) test_calc_disc_values_error() (umami.tests.unit.metrics.test_metrics.CalcDiscValuesTestCase method) test_calc_disc_values_rejection() (umami.tests.unit.metrics.test_metrics.CalcDiscValuesTestCase method) test_calculate_fraction_dict() (umami.tests.unit.evaluation_tools.test_eval_tools.SmallFunctionsTestCase method) test_calculate_fraction_dict_value_error() (umami.tests.unit.evaluation_tools.test_eval_tools.SmallFunctionsTestCase method) test_call() (umami.tests.unit.tf_tools.test_generators.TestCadsGenerator method) (umami.tests.unit.tf_tools.test_generators.TestDipsGenerator method) (umami.tests.unit.tf_tools.test_generators.TestDL1Generator method) (umami.tests.unit.tf_tools.test_generators.TestUmamiCondAttGenerator method) (umami.tests.unit.tf_tools.test_generators.TestUmamiGenerator method) (umami.tests.unit.tf_tools.test_tf_tools.AttentionTestCase method) (umami.tests.unit.tf_tools.test_tf_tools.ConditionalDeepSetTestCase method) (umami.tests.unit.tf_tools.test_tf_tools.DeepSetTestCase method) (umami.tests.unit.tf_tools.test_tf_tools.DenseNetTestCase method) test_call_assertion_error() (umami.tests.unit.tf_tools.test_tf_tools.AttentionTestCase method) (umami.tests.unit.tf_tools.test_tf_tools.DeepSetTestCase method) test_call_no_batch_norm() (umami.tests.unit.tf_tools.test_tf_tools.DeepSetTestCase method) test_call_no_softmax() (umami.tests.unit.tf_tools.test_tf_tools.AttentionTestCase method) test_call_with_mask() (umami.tests.unit.tf_tools.test_tf_tools.AttentionTestCase method) (umami.tests.unit.tf_tools.test_tf_tools.DeepSetTestCase method) test_call_without_weights() (umami.tests.unit.tf_tools.test_generators.TestDipsGenerator method) (umami.tests.unit.tf_tools.test_generators.TestDL1Generator method) test_check_main_class_input_fail() (umami.tests.unit.helper_tools.test_tools.CheckMainClassInputTestCase method) test_check_main_class_input_list() (umami.tests.unit.helper_tools.test_tools.CheckMainClassInputTestCase method) test_check_main_class_input_set() (umami.tests.unit.helper_tools.test_tools.CheckMainClassInputTestCase method) test_check_main_class_input_str() (umami.tests.unit.helper_tools.test_tools.CheckMainClassInputTestCase method) test_circular_modelgen_eq_tddgen() (umami.tests.unit.tf_tools.test_tddgenerators.TestTDDGenerator method) test_circular_modelgen_eq_tddgen_old_scaling() (umami.tests.unit.tf_tools.test_tddgenerators.TestTDDGenerator method) test_circular_with_excluded_var() (umami.tests.unit.tf_tools.test_tddgenerators.TestTDDGenerator method) test_circular_with_weights_in_tdd() (umami.tests.unit.tf_tools.test_tddgenerators.TestTDDGenerator method) test_common_vars() (umami.tests.unit.data_tools.test_data_tools_tools.CompareH5FilesVariablesTestCase method) test_common_vars_3_files() (umami.tests.unit.data_tools.test_data_tools_tools.CompareH5FilesVariablesTestCase method) test_compare_leading_spaces() (in module umami.tests.unit.helper_tools.test_tools) test_compute_mask() (umami.tests.unit.tf_tools.test_tf_tools.AttentionTestCase method) (umami.tests.unit.tf_tools.test_tf_tools.DeepSetTestCase method) test_compute_mask_errors() (umami.tests.unit.tf_tools.test_tf_tools.AttentionTestCase method) test_compute_mask_no_mask() (umami.tests.unit.tf_tools.test_tf_tools.ConditionalDeepSetTestCase method) test_correct_config_dips() (umami.tests.unit.train_tools.test_nn_tools.GetDropoutRatesTestCase method) test_correct_output() (umami.tests.unit.preprocessing.test_preprocessing_tools.BinariseJetLabelsTestCase method) test_count_no_samples_defined() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.UnderSamplingTestCase method) test_create_from_dataset() (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.CreateDatasetsTestCase method) test_create_from_dict() (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.CreateDatasetsTestCase method) test_create_metadata_folder() (umami.tests.unit.train_tools.test_nn_tools.CreateMetadataFolderTestCase method) test_create_metadata_folder_overwrite() (umami.tests.unit.train_tools.test_nn_tools.CreateMetadataFolderTestCase method) test_create_samples_dict() (umami.tests.unit.preprocessing.test_preprocessing_tools.CreateSamplesTestCase method) test_cut_operators() (umami.tests.unit.data_tools.test_data_tools_loaders.LoadFilesTestCase method) test_cut_vars() (umami.tests.unit.data_tools.test_data_tools_loaders.LoadFilesTestCase method) test_cuts_passing_ttbar() (umami.tests.unit.data_tools.test_data_tools_cuts.GetSampleCutsTestCase method) test_cuts_passing_ttbar_extended_labelling() (umami.tests.unit.data_tools.test_data_tools_cuts.GetSampleCutsTestCase method) test_cuts_wrong_format() (umami.tests.unit.data_tools.test_data_tools_cuts.GetSampleCutsTestCase method) test_cuts_wrong_modulo() (umami.tests.unit.data_tools.test_data_tools_cuts.GetSampleCutsTestCase method) test_cuts_wrong_operator() (umami.tests.unit.data_tools.test_data_tools_cuts.GetSampleCutsTestCase method) test_default_batch_size() (umami.tests.unit.preprocessing.test_preprocessing_tools_configuration.PreparationTestCase method) test_defaults_vars() (umami.tests.unit.preprocessing.test_preprocessing_tools.GetVariableDictTestCase method) test_denominator_float() (umami.tests.unit.helper_tools.test_histogram_tools.SaveDivideTestCase method) test_different_chunk_size() (umami.tests.unit.data_tools.test_data_tools_loaders.LoadFilesTestCase method) test_different_input_length_class_names() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.CorrectFractionsTestCase method) test_different_input_lengths() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.CorrectFractionsTestCase method) test_different_keys_case() (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.CheckKeysTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.CheckShapesTestCase method) test_different_samples_per_category() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.UnderSamplingNoReplaceTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.UnderSamplingTestCase method) test_different_size_case() (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.CheckSizeTestCase method) test_different_vars() (umami.tests.unit.data_tools.test_data_tools_tools.CompareH5FilesVariablesTestCase method) test_different_vars_3_files() (umami.tests.unit.data_tools.test_data_tools_tools.CompareH5FilesVariablesTestCase method) test_dips_att_with_cond_info() (umami.tests.unit.train_tools.test_nn_tools.ConfigurationTestCase method) test_dips_config() (umami.tests.unit.train_tools.test_nn_tools.ConfigurationTestCase method) test_discriminant_output_shape() (umami.tests.unit.metrics.test_metrics.MetricsSmallTestCase method) test_double_defined_b_jets() (umami.tests.unit.train_tools.test_nn_tools.ConfigurationTestCase method) test_double_defined_c_jets() (umami.tests.unit.train_tools.test_nn_tools.ConfigurationTestCase method) test_double_label_value() (umami.tests.unit.train_tools.test_nn_tools.ConfigurationTestCase method) test_double_neested() (umami.tests.unit.tools.test_tools.FlattenListTest method) test_double_output_name() (umami.tests.unit.preprocessing.test_preprocessing_tools_configuration.PreparationTestCase method) test_duplicate_error() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.SamplingGeneratorTestCase method) test_empty() (umami.tests.unit.tools.test_tools.FlattenListTest method) test_equal_fractions() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.UnderSamplingNoReplaceTestCase method) test_equal_keys_case() (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.CheckKeysTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.CheckShapesTestCase method) test_equal_length() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.UnderSamplingTestCaseEasyDistr method) test_equal_length_smalln() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.UnderSamplingTestCase method) test_equal_size_case() (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.CheckSizeTestCase method) test_evaluate_tagger_in_files() (umami.tests.integration.test_train.TestTraining method) test_event_indices() (umami.tests.unit.preprocessing.test_preprocessing_tools.TTbarMergeTestCase method) test_event_list() (umami.tests.unit.preprocessing.test_preprocessing_tools.TTbarMergeTestCase method) test_example_plots() (in module umami.tests.integration.test_examples) test_exclude_in_header() (umami.tests.unit.train_tools.test_nn_tools.GetJetFeatureIndicesTestCase method) test_failed_get_sample() (umami.tests.unit.preprocessing.test_preprocessing_tools_configuration.PreparationTestCase method) test_faulty_setup() (umami.tests.unit.tf_tools.test_tf_tools_convert_to_record.ConvertTest method) test_files (umami.train_tools.configuration.TrainConfigurationObject attribute) test_float_case() (umami.tests.unit.data_tools.test_data_tools_cuts.GetCategoryCutsTestCase method) test_genera_settings_class_plot_options_as_dict() (umami.tests.unit.preprocessing.test_preprocessing_tools_configuration.PreprocessConfigurationTestCase method) test_general_settings_class_as_dict() (umami.tests.unit.preprocessing.test_preprocessing_tools_configuration.PreprocessConfigurationTestCase method) test_get_batches_per_file() (umami.tests.unit.preprocessing.test_preprocessing_tools.PrepareSamplesTestCase method) test_get_class_label_variables_3_classes() (umami.tests.unit.helper_tools.test_classification_tools.GetClassTestCase method) test_get_class_label_variables_4_classes() (umami.tests.unit.helper_tools.test_classification_tools.GetClassTestCase method) test_get_class_prob_var_names_3_classes() (umami.tests.unit.helper_tools.test_classification_tools.GetClassTestCase method) test_get_class_prob_var_names_4_classes() (umami.tests.unit.helper_tools.test_classification_tools.GetClassTestCase method) test_get_comp_tagger_rej_dict_dips_dl1() (umami.tests.unit.plotting_tools.test_train_plot_functions.GetCompTaggerRejDictTestCase method) test_get_comp_tagger_rej_dict_errors() (umami.tests.unit.plotting_tools.test_train_plot_functions.GetCompTaggerRejDictTestCase method) test_get_comp_tagger_rej_dict_umami() (umami.tests.unit.plotting_tools.test_train_plot_functions.GetCompTaggerRejDictTestCase method) test_get_config() (umami.tests.unit.tf_tools.test_tf_tools.AttentionTestCase method) (umami.tests.unit.tf_tools.test_tf_tools.ConditionalDeepSetTestCase method) (umami.tests.unit.tf_tools.test_tf_tools.DeepSetTestCase method) (umami.tests.unit.tf_tools.test_tf_tools.DenseNetTestCase method) test_get_cut_list_3_classes() (umami.tests.unit.helper_tools.test_classification_tools.GetClassTestCase method) test_get_cut_list_4_classes() (umami.tests.unit.helper_tools.test_classification_tools.GetClassTestCase method) test_get_datasets_configuration_all_default() (umami.tests.unit.input_vars_tools.test_input_vars_tools.HelperFunctionTestCase method) test_get_datasets_configuration_specific_class_labels() (umami.tests.unit.input_vars_tools.test_input_vars_tools.HelperFunctionTestCase method) test_get_dict_name() (umami.tests.unit.train_tools.test_nn_tools.GetMetricsFileNameTestCase method) test_get_dict_name_without_val_dict_path() (umami.tests.unit.train_tools.test_nn_tools.GetMetricsFileNameTestCase method) test_get_epoch_from_string() (umami.tests.unit.train_tools.test_nn_tools.GetEpochFromStringTestCase method) test_get_file_name_no_input() (umami.tests.unit.preprocessing.test_preprocessing_tools_configuration.PreprocessConfigurationTestCase method) test_get_file_name_no_iterations() (umami.tests.unit.preprocessing.test_preprocessing_tools_configuration.PreprocessConfigurationTestCase method) test_get_file_name_no_iterations_no_input() (umami.tests.unit.preprocessing.test_preprocessing_tools_configuration.PreprocessConfigurationTestCase method) test_get_generator() (umami.tests.unit.tf_tools.test_tddgenerators.TestTDDGenerator method) test_get_indices() (umami.tests.unit.preprocessing.test_preprocessing_tools.TTbarMergeTestCase method) test_get_input_files() (umami.tests.unit.preprocessing.test_preprocessing_tools.TTbarMergeTestCase method) test_get_jet_feature_indices() (umami.tests.unit.train_tools.test_nn_tools.GetJetFeatureIndicesTestCase method) test_get_jet_feature_position() (umami.tests.unit.train_tools.test_nn_tools.GetJetFeaturePositionTestCase method) test_get_jet_feature_position_value_error() (umami.tests.unit.train_tools.test_nn_tools.GetJetFeaturePositionTestCase method) test_get_model_path() (umami.tests.unit.train_tools.test_nn_tools.GetModelPathTestCase method) test_get_parameters() (umami.tests.unit.train_tools.test_nn_tools.GetMetricsFileNameTestCase method) test_get_parameters_exception() (umami.tests.unit.train_tools.test_nn_tools.GetMetricsFileNameTestCase method) test_get_rejection() (umami.tests.unit.metrics.test_metrics.GetRejectionTestCase method) test_get_rejection_4_classes() (umami.tests.unit.metrics.test_metrics.GetRejectionTestCase method) test_get_rejection_wrong_shapes() (umami.tests.unit.metrics.test_metrics.GetRejectionTestCase method) test_get_scale_dict() (umami.tests.unit.preprocessing.test_preprocessing_tools.GetDictTestCase method) test_get_scaling() (umami.tests.unit.preprocessing.test_preprocessing_tools_scaling.ScalingTestCase method) test_get_scaling_tracks() (umami.tests.unit.preprocessing.test_preprocessing_tools_scaling.ScalingTestCase method) test_get_score() (umami.tests.unit.metrics.test_metrics.GetScoreTestCase method) test_get_score_4classes() (umami.tests.unit.metrics.test_metrics.GetScoreTestCase method) test_get_score_errors() (umami.tests.unit.metrics.test_metrics.GetScoreTestCase method) test_get_test_file() (umami.tests.unit.train_tools.test_nn_tools.GetSamplesTestCase method) test_get_test_sample() (umami.tests.unit.train_tools.test_nn_tools.GetSamplesTestCase method) test_get_test_sample_errors() (umami.tests.unit.train_tools.test_nn_tools.GetSamplesTestCase method) test_get_test_sample_extended_labeling() (umami.tests.unit.train_tools.test_nn_tools.GetSamplesTestCase method) test_get_test_sample_trks() (umami.tests.unit.train_tools.test_nn_tools.GetSamplesTestCase method) test_get_test_sample_trks_extended_labeling() (umami.tests.unit.train_tools.test_nn_tools.GetSamplesTestCase method) test_get_variable_dict() (umami.tests.unit.preprocessing.test_preprocessing_tools.GetDictTestCase method) test_h5_errors() (umami.tests.unit.data_tools.test_data_tools_loaders.LoadFilesTestCase method) test_half_case() (umami.tests.unit.helper_tools.test_histogram_tools.SaveDivideTestCase method) test_hardcoded_dict() (umami.tests.unit.train_tools.test_nn_tools.GetUniqueIdentifiersTestCase method) test_hist_not_same_length_denomiantor_and_unc() (umami.tests.unit.helper_tools.test_histogram_tools.HistRatioTestCase method) test_hist_not_same_length_nomiantor_and_unc() (umami.tests.unit.helper_tools.test_histogram_tools.HistRatioTestCase method) test_hist_not_same_length_nominator_denominator() (umami.tests.unit.helper_tools.test_histogram_tools.HistRatioTestCase method) test_hist_ratio() (umami.tests.unit.helper_tools.test_histogram_tools.HistRatioTestCase method) test_hist_w_unc_normed() (umami.tests.unit.helper_tools.test_histogram_tools.HistWUncTestCase method) test_hist_w_unc_not_normed() (umami.tests.unit.helper_tools.test_histogram_tools.HistWUncTestCase method) test_hist_w_unc_zero_case() (umami.tests.unit.helper_tools.test_histogram_tools.HistWUncTestCase method) test_init() (umami.tests.unit.tf_tools.test_generators.TestModelGenerator method) test_init_fresh_model() (umami.tests.unit.tf_tools.test_tf_tools.PrepareModelTestCase method) test_init_fresh_model_no_load_optimiser_given() (umami.tests.unit.tf_tools.test_tf_tools.PrepareModelTestCase method) test_init_no_continue() (umami.tests.unit.train_tools.test_nn_tools.CallbackBaseTestCase method) test_init_no_n_jets_no_x_given() (umami.tests.unit.tf_tools.test_generators.TestModelGenerator method) test_init_no_n_jets_x_name_given() (umami.tests.unit.tf_tools.test_generators.TestModelGenerator method) test_init_no_n_jets_x_trk_name_given() (umami.tests.unit.tf_tools.test_generators.TestModelGenerator method) test_init_no_val_data() (umami.tests.unit.train_tools.test_nn_tools.CallbackBaseTestCase method) test_init_with_continue_new_file_init() (umami.tests.unit.train_tools.test_nn_tools.CallbackBaseTestCase method) test_init_with_continue_no_file_init() (umami.tests.unit.train_tools.test_nn_tools.CallbackBaseTestCase method) test_input_correct_fractions() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.CorrectFractionsTestCase method)

test_integer_case() (umami.tests.unit.data_tools.test_data_tools_cuts.GetCategoryCutsTestCase method) test_jets_generator_fullcuts() (umami.tests.unit.preprocessing.test_preprocessing_tools.PrepareSamplesTestCase method) test_jets_generator_fullcuts_wotracks() (umami.tests.unit.preprocessing.test_preprocessing_tools.PrepareSamplesTestCase method) test_jets_generator_lightcut() (umami.tests.unit.preprocessing.test_preprocessing_tools.PrepareSamplesTestCase method) test_join_scale_dicts() (umami.tests.unit.preprocessing.test_preprocessing_tools_scaling.ScalingTestCase method) test_join_scale_dicts_trks() (umami.tests.unit.preprocessing.test_preprocessing_tools_scaling.ScalingTestCase method) test_key_error_scale_dict() (umami.tests.unit.preprocessing.test_preprocessing_tools_scaling.ApplyScalingJetsTestCase method) test_label_reading() (umami.tests.unit.preprocessing.test_preprocessing_tools.GetVariableDictTestCase method) test_line() (umami.tests.unit.helper_tools.test_tools.ReplaceLineInFileTestCase method) test_line_fail() (umami.tests.unit.helper_tools.test_tools.ReplaceLineInFileTestCase method) test_list_case() (umami.tests.unit.data_tools.test_data_tools_cuts.GetCategoryCutsTestCase method) test_load_events_generator() (umami.tests.unit.preprocessing.test_preprocessing_tools.TTbarMergeTestCase method) test_load_in_memory() (umami.tests.unit.tf_tools.test_generators.TestModelGenerator method) test_load_in_memory_x_name_error() (umami.tests.unit.tf_tools.test_generators.TestModelGenerator method) test_load_in_memory_x_trk_name_error() (umami.tests.unit.tf_tools.test_generators.TestModelGenerator method) test_load_jets_from_file() (umami.tests.unit.data_tools.test_data_tools_loaders.LoadFilesTestCase method) test_load_model_with_continue_training() (umami.tests.unit.tf_tools.test_tf_tools.PrepareModelTestCase method) test_load_model_without_continue_training() (umami.tests.unit.tf_tools.test_tf_tools.PrepareModelTestCase method) test_load_optimiser_value_rrror() (umami.tests.unit.tf_tools.test_tf_tools.PrepareModelTestCase method) test_load_trks_from_file() (umami.tests.unit.data_tools.test_data_tools_loaders.LoadFilesTestCase method) test_load_validation_data() (umami.tests.unit.train_tools.test_nn_tools.GetSamplesTestCase method) test_load_validation_data_no_var_cuts() (umami.tests.unit.train_tools.test_nn_tools.GetSamplesTestCase method) test_load_validation_data_unsupported_tagger() (umami.tests.unit.train_tools.test_nn_tools.GetSamplesTestCase method) test_missing_internal_label() (umami.tests.unit.preprocessing.test_preprocessing_tools.BinariseJetLabelsTestCase method) test_missing_key_error() (umami.tests.unit.preprocessing.test_preprocessing_tools_configuration.PreprocessConfigurationTestCase method) (umami.tests.unit.train_tools.test_nn_tools.ConfigurationTestCase method) test_multiple_lines() (umami.tests.unit.helper_tools.test_tools.ReplaceLineInFileTestCase method) test_multiple_lines_fail() (umami.tests.unit.helper_tools.test_tools.ReplaceLineInFileTestCase method) test_my_callback() (umami.tests.unit.train_tools.test_nn_tools.MyCallbackTestCase method) test_my_callback_umami() (umami.tests.unit.train_tools.test_nn_tools.MyCallbackUmamiTestCase method) test_neested_with_dict() (umami.tests.unit.tools.test_tools.FlattenListTest method) test_nested_list_case() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.CalculateBinningTestCase static method) test_nested_structure() (umami.tests.unit.preprocessing.test_preprocessing_tools.GetVariableDictTestCase method) test_no_batch_size() (umami.tests.unit.train_tools.test_nn_tools.ConfigurationTestCase method) test_no_duplicate() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.SamplingGeneratorTestCase method) test_no_duplicate_no_tracks() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.SamplingGeneratorTestCase method) test_no_eval_batch_size() (umami.tests.unit.train_tools.test_nn_tools.ConfigurationTestCase method) test_no_inputs() (umami.tests.unit.data_tools.test_data_tools_tools.CompareH5FilesVariablesTestCase method) test_no_sample_provided() (umami.tests.unit.preprocessing.test_preprocessing_tools.PrepareSamplesTestCase method) test_no_samples_defined() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.PDFResamplingTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.UnderSamplingNoReplaceTestCase method) test_no_tracks() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.SamplingGeneratorTestCase method) test_no_val_batch_size() (umami.tests.unit.train_tools.test_nn_tools.ConfigurationTestCase method) test_no_val_no_eval_batch_size() (umami.tests.unit.train_tools.test_nn_tools.ConfigurationTestCase method) test_non_list_case() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.CalculateBinningTestCase method) test_none() (umami.tests.unit.tools.test_tools.FlattenListTest method) test_not_fraction_sum_one() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.CorrectFractionsTestCase method) test_numerator_float() (umami.tests.unit.helper_tools.test_histogram_tools.SaveDivideTestCase method) test_ones_case() (umami.tests.unit.helper_tools.test_histogram_tools.SaveDivideTestCase method) test_parser() (umami.tests.unit.preprocessing.test_preprocessing.PreprocessingTestParser method) test_plot_accuracies() (umami.tests.unit.plotting_tools.test_train_plot_functions.TrainPlotsTestCase method) test_plot_accuracies_umami() (umami.tests.unit.plotting_tools.test_train_plot_functions.TrainPlotsTestCase method) test_plot_disc_cut_per_epoch() (umami.tests.unit.plotting_tools.test_train_plot_functions.TrainPlotsTestCase method) test_plot_disc_cut_per_epoch_umami() (umami.tests.unit.plotting_tools.test_train_plot_functions.TrainPlotsTestCase method) test_plot_fraction_contour() (umami.tests.unit.plotting_tools.test_eval_plot_functions.EvalPlotsTestCase method) test_plot_input_vars() (umami.tests.integration.test_input_vars_plot.TestInputVarsPlotting method) test_plot_input_vars_jets() (umami.tests.unit.input_vars_tools.test_input_vars_tools.JetPlottingTestCase method) test_plot_input_vars_jets_comparison() (umami.tests.unit.input_vars_tools.test_input_vars_tools.JetPlottingTestCase method) test_plot_input_vars_jets_comparison_wrong_type() (umami.tests.unit.input_vars_tools.test_input_vars_tools.JetPlottingTestCase method) test_plot_input_vars_jets_log_value_error() (umami.tests.unit.input_vars_tools.test_input_vars_tools.JetPlottingTestCase method) test_plot_input_vars_jets_wrong_type() (umami.tests.unit.input_vars_tools.test_input_vars_tools.JetPlottingTestCase method) test_plot_input_vars_trks() (umami.tests.unit.input_vars_tools.test_input_vars_tools.JetPlottingTestCase method) test_plot_input_vars_trks_comparison() (umami.tests.unit.input_vars_tools.test_input_vars_tools.JetPlottingTestCase method) test_plot_input_vars_trks_comparison_not_normalised() (umami.tests.unit.input_vars_tools.test_input_vars_tools.JetPlottingTestCase method) test_plot_input_vars_trks_comparison_wrong_type() (umami.tests.unit.input_vars_tools.test_input_vars_tools.JetPlottingTestCase method) test_plot_input_vars_trks_log_error_raise() (umami.tests.unit.input_vars_tools.test_input_vars_tools.JetPlottingTestCase method) test_plot_input_vars_trks_wrong_type() (umami.tests.unit.input_vars_tools.test_input_vars_tools.JetPlottingTestCase method) test_plot_losses() (umami.tests.unit.plotting_tools.test_train_plot_functions.TrainPlotsTestCase method) test_plot_losses_umami() (umami.tests.unit.plotting_tools.test_train_plot_functions.TrainPlotsTestCase method) test_plot_n_tracks_per_jet() (umami.tests.unit.input_vars_tools.test_input_vars_tools.JetPlottingTestCase method) test_plot_prob() (umami.tests.unit.plotting_tools.test_eval_plot_functions.EvalPlotsTestCase method) test_plot_prob_comparison() (umami.tests.unit.plotting_tools.test_eval_plot_functions.EvalPlotsTestCase method) test_plot_pt_dependence() (umami.tests.unit.plotting_tools.test_eval_plot_functions.EvalPlotsTestCase method) test_plot_rej_per_epoch() (umami.tests.unit.plotting_tools.test_train_plot_functions.TrainPlotsTestCase method) test_plot_rej_per_epoch_comp() (umami.tests.unit.plotting_tools.test_train_plot_functions.TrainPlotsTestCase method) test_plot_resampling_variables() (umami.tests.unit.plotting_tools.test_preprocessing_plotting_functions.PlotResamplingVariablesTestCase method) test_plot_resampling_variables_wrong_binning() (umami.tests.unit.plotting_tools.test_preprocessing_plotting_functions.PlotResamplingVariablesTestCase method) test_plot_roc() (umami.tests.unit.plotting_tools.test_eval_plot_functions.EvalPlotsTestCase method) test_plot_roc_comparison() (umami.tests.unit.plotting_tools.test_eval_plot_functions.EvalPlotsTestCase method) test_plot_saliency() (umami.tests.unit.plotting_tools.test_eval_plot_functions.EvalPlotsTestCase method) test_plot_score() (umami.tests.unit.plotting_tools.test_eval_plot_functions.EvalPlotsTestCase method) test_plot_score_comparison() (umami.tests.unit.plotting_tools.test_eval_plot_functions.EvalPlotsTestCase method) test_plot_unique_jet_appearence() (umami.tests.unit.plotting_tools.test_preprocessing_plotting_functions.PreprocessingPlotsTestCase method) test_plotting_umami_dips() (umami.tests.integration.test_plotting_umami.TestPlottingUmami method) test_plotting_umami_dl1r() (umami.tests.integration.test_plotting_umami.TestPlottingUmami method) test_plotting_umami_umami() (umami.tests.integration.test_plotting_umami.TestPlottingUmami method) test_preprocessing_additional_jet_labels() (umami.tests.integration.test_preprocessing.TestPreprocessing method) test_preprocessing_dips_count() (umami.tests.integration.test_preprocessing.TestPreprocessing method) test_preprocessing_dips_four_classes_pdf() (umami.tests.integration.test_preprocessing.TestPreprocessing method) test_preprocessing_dips_hits_count() (umami.tests.integration.test_preprocessing.TestPreprocessing method) test_preprocessing_dips_pdf() (umami.tests.integration.test_preprocessing.TestPreprocessing method) test_preprocessing_dips_upp_countup() (umami.tests.integration.test_preprocessing_upp.TestPreprocessingUPP method) test_preprocessing_dips_upp_pdf() (umami.tests.integration.test_preprocessing_upp.TestPreprocessingUPP method) test_preprocessing_dips_weighting() (umami.tests.integration.test_preprocessing.TestPreprocessing method) test_preprocessing_dl1r_count() (umami.tests.integration.test_preprocessing.TestPreprocessing method) test_preprocessing_dl1r_pdf() (umami.tests.integration.test_preprocessing.TestPreprocessing method) test_preprocessing_dl1r_weighting() (umami.tests.integration.test_preprocessing.TestPreprocessing method) test_preprocessing_plots() (umami.tests.unit.plotting_tools.test_preprocessing_plotting_functions.PreprocessingPlotsTestCase method) test_preprocessing_umami_count() (umami.tests.integration.test_preprocessing.TestPreprocessing method) test_preprocessing_umami_importance_no_replace() (umami.tests.integration.test_preprocessing.TestPreprocessing method) test_preprocessing_umami_pdf() (umami.tests.integration.test_preprocessing.TestPreprocessing method) test_preprocessing_umami_weighting() (umami.tests.integration.test_preprocessing.TestPreprocessing method) test_preprocessing_upp_flags() (umami.tests.integration.test_preprocessing_upp.TestPreprocessingUPP method) test_run() (umami.tests.unit.preprocessing.test_preprocessing_tools.PrepareSamplesTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.PDFResamplingTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.UnderSamplingTestCase method) test_sampling_class_as_dict() (umami.tests.unit.preprocessing.test_preprocessing_tools_configuration.PreprocessConfigurationTestCase method) test_sampling_options_class_as_dict() (umami.tests.unit.preprocessing.test_preprocessing_tools_configuration.PreprocessConfigurationTestCase method) test_save_parameters() (umami.tests.unit.tf_tools.test_tf_tools_convert_to_record.ConvertTest method) test_save_parameters_nadd_vars() (umami.tests.unit.tf_tools.test_tf_tools_convert_to_record.ConvertTest method) test_scaling_down_largest() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.CorrectFractionsTestCase method) test_scaling_down_small() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.CorrectFractionsTestCase method) test_scaling_jets() (umami.tests.unit.preprocessing.test_preprocessing_tools_scaling.ApplyScalingJetsTestCase method) test_scaling_structured_array_jets() (umami.tests.unit.preprocessing.test_preprocessing_tools_scaling.ApplyScalingJetsTestCase method) test_setup_output_directory() (umami.tests.unit.train_tools.test_nn_tools.SetupOutputDirectoryTestCase method) test_setup_output_directory_clean() (umami.tests.unit.train_tools.test_nn_tools.SetupOutputDirectoryTestCase method) test_setup_output_directory_continue() (umami.tests.unit.train_tools.test_nn_tools.SetupOutputDirectoryTestCase method) test_setup_output_directory_error() (umami.tests.unit.train_tools.test_nn_tools.SetupOutputDirectoryTestCase method) test_shape_2_classes() (umami.tests.unit.preprocessing.test_preprocessing_tools.BinariseJetLabelsTestCase method) test_shape_array() (umami.tests.unit.preprocessing.test_preprocessing_tools.BinariseJetLabelsTestCase method) test_shape_data_frame() (umami.tests.unit.preprocessing.test_preprocessing_tools.BinariseJetLabelsTestCase method) test_simple_neested() (umami.tests.unit.tools.test_tools.FlattenListTest method) test_single_list_case() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.CalculateBinningTestCase static method) test_total_size() (umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5.GetSizeTestCase method) test_tracks() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.SamplingGeneratorTestCase method) test_train_cads() (umami.tests.integration.test_train.TestTraining method) test_train_cond_att_umami() (umami.tests.integration.test_train.TestTraining method) test_train_dips_four_classes() (umami.tests.integration.test_train.TestTraining method) test_train_dips_no_attention() (umami.tests.integration.test_train.TestTraining method) test_train_dl1r() (umami.tests.integration.test_train.TestTraining method) test_train_tfrecords_cads() (umami.tests.integration.test_train.TestTraining method) test_train_tfrecords_cond_att_umami() (umami.tests.integration.test_train.TestTraining method) test_train_tfrecords_dips() (umami.tests.integration.test_train.TestTraining method) test_train_tfrecords_dl1r() (umami.tests.integration.test_train.TestTraining method) test_train_tfrecords_umami() (umami.tests.integration.test_train.TestTraining method) test_train_umami() (umami.tests.integration.test_train.TestTraining method) test_train_variables() (umami.tests.unit.preprocessing.test_preprocessing_tools.GetVariableDictTestCase method) test_trk_train_variables() (umami.tests.unit.preprocessing.test_preprocessing_tools.GetVariableDictTestCase method) test_trks_different_chunk_size() (umami.tests.unit.data_tools.test_data_tools_loaders.LoadFilesTestCase method) test_trks_different_operators() (umami.tests.unit.data_tools.test_data_tools_loaders.LoadFilesTestCase method) test_trks_raise_errors() (umami.tests.unit.data_tools.test_data_tools_loaders.LoadFilesTestCase method) test_twice_same_fractions() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.CorrectFractionsTestCase method) test_umami_config() (umami.tests.unit.train_tools.test_nn_tools.ConfigurationTestCase method) test_value_error_scale_value() (umami.tests.unit.preprocessing.test_preprocessing_tools_scaling.ApplyScalingJetsTestCase method) test_wrong_category_provided() (umami.tests.unit.preprocessing.test_preprocessing_tools.PrepareSamplesTestCase method) test_wrong_filepath() (umami.tests.unit.data_tools.test_data_tools_loaders.LoadFilesTestCase method) test_wrong_length() (umami.tests.unit.train_tools.test_nn_tools.GetDropoutRatesTestCase method) test_wrong_sample_provided() (umami.tests.unit.preprocessing.test_preprocessing_tools.PrepareSamplesTestCase method) test_wrong_type() (umami.tests.unit.preprocessing.test_preprocessing_tools.BinariseJetLabelsTestCase method) test_wrong_type_provided() (umami.tests.unit.data_tools.test_data_tools_cuts.GetCategoryCutsTestCase method) test_zero_case() (umami.tests.unit.helper_tools.test_histogram_tools.SaveDivideTestCase method) test_zero_length() (umami.tests.unit.preprocessing.test_preprocessing_tools.BinariseJetLabelsTestCase method) (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.CorrectFractionsTestCase method) test_zero_n_jets() (umami.tests.unit.preprocessing.test_preprocessing_tools_resampling.CorrectFractionsTestCase method) TestCadsGenerator (class in umami.tests.unit.tf_tools.test_generators) TestDipsGenerator (class in umami.tests.unit.tf_tools.test_generators) TestDL1Generator (class in umami.tests.unit.tf_tools.test_generators) TestInputVarsPlotting (class in umami.tests.integration.test_input_vars_plot) TestModelGenerator (class in umami.tests.unit.tf_tools.test_generators) TestPlottingUmami (class in umami.tests.integration.test_plotting_umami) TestPreprocessing (class in umami.tests.integration.test_preprocessing) TestPreprocessingUPP (class in umami.tests.integration.test_preprocessing_upp) TestTDDGenerator (class in umami.tests.unit.tf_tools.test_tddgenerators) TestTraining (class in umami.tests.integration.test_train) TestUmamiCondAttGenerator (class in umami.tests.unit.tf_tools.test_generators) TestUmamiGenerator (class in umami.tests.unit.tf_tools.test_generators) TFRecordReader (class in umami.tf_tools.load_tfrecord) tracks_name (umami.train_tools.configuration.TrainConfigurationObject attribute) tracks_names (umami.preprocessing_tools.configuration.SamplingOptions attribute) train_cads() (in module umami.models.model_cads) train_data_structure (umami.train_tools.configuration.TrainConfigurationObject attribute) train_dips() (in module umami.models.model_dips) train_dl1() (in module umami.models.model_dl1) train_file (umami.train_tools.configuration.TrainConfigurationObject attribute) train_umami() (in module umami.models.model_umami) train_umami_cond_att() (in module umami.models.model_umami_cond_att) TrainConfiguration (class in umami.train_tools.configuration) TrainConfigurationObject (class in umami.train_tools.configuration) trained_taggers (umami.train_tools.configuration.ValidationSettingsConfig attribute) TrainPlotsTestCase (class in umami.tests.unit.plotting_tools.test_train_plot_functions) TrainSampleWriter (class in umami.preprocessing_tools.writing_train_file) translate_binning() (in module umami.plotting_tools.utils) translate_kwargs() (in module umami.plotting_tools.utils) TTbarMerge (class in umami.preprocessing_tools.ttbar_merge) TTbarMergeTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools) type (umami.preprocessing_tools.configuration.Sample attribute)

U

umami module umami.configuration module umami.configuration.base module umami.configuration.configuration module umami.data_tools module umami.data_tools.cuts module umami.data_tools.loaders module umami.data_tools.tools module umami.evaluate_model module umami.evaluation_tools module umami.evaluation_tools.eval_tools module umami.evaluation_tools.feature_importance module umami.helper_tools module umami.helper_tools.classification_tools module umami.helper_tools.histogram_tools module umami.input_vars_tools module umami.input_vars_tools.plotting_functions module umami.metrics module umami.metrics.metrics module umami.models module umami.models.model_cads module umami.models.model_dips module umami.models.model_dl1 module umami.models.model_umami module umami.models.model_umami_cond_att module umami.plot_input_variables module umami.plotting_epoch_performance module umami.plotting_tools module umami.plotting_tools.eval_plotting_functions module umami.plotting_tools.preprocessing_plotting_functions module umami.plotting_tools.train_plotting_functions module umami.plotting_tools.utils module umami.plotting_umami module umami.preprocessing module umami.preprocessing_tools module umami.preprocessing_tools.configuration module umami.preprocessing_tools.merging module umami.preprocessing_tools.preparation module umami.preprocessing_tools.resampling module umami.preprocessing_tools.resampling.count_sampling module umami.preprocessing_tools.resampling.importance_sampling_no_replace module umami.preprocessing_tools.resampling.pdf_sampling module umami.preprocessing_tools.resampling.resampling_base module umami.preprocessing_tools.resampling.weighting module umami.preprocessing_tools.scaling module umami.preprocessing_tools.ttbar_merge module umami.preprocessing_tools.utils module umami.preprocessing_tools.writing_train_file module umami.sample_merging module umami.tests module umami.tests.integration module umami.tests.integration.test_examples module umami.tests.integration.test_input_vars_plot module umami.tests.integration.test_plotting_umami module umami.tests.integration.test_preprocessing module umami.tests.integration.test_preprocessing_upp module umami.tests.integration.test_train module

umami.tests.unit module umami.tests.unit.data_tools module umami.tests.unit.data_tools.test_data_tools_cuts module umami.tests.unit.data_tools.test_data_tools_loaders module umami.tests.unit.data_tools.test_data_tools_tools module umami.tests.unit.evaluation_tools module umami.tests.unit.evaluation_tools.test_eval_tools module umami.tests.unit.helper_tools module umami.tests.unit.helper_tools.test_classification_tools module umami.tests.unit.helper_tools.test_histogram_tools module umami.tests.unit.helper_tools.test_tools module umami.tests.unit.input_vars_tools module umami.tests.unit.input_vars_tools.test_input_vars_tools module umami.tests.unit.metrics module umami.tests.unit.metrics.test_metrics module umami.tests.unit.plotting_tools module umami.tests.unit.plotting_tools.test_eval_plot_functions module umami.tests.unit.plotting_tools.test_plotting_utils module umami.tests.unit.plotting_tools.test_preprocessing_plotting_functions module umami.tests.unit.plotting_tools.test_train_plot_functions module umami.tests.unit.preprocessing module umami.tests.unit.preprocessing.test_preprocessing module umami.tests.unit.preprocessing.test_preprocessing_tools module umami.tests.unit.preprocessing.test_preprocessing_tools_configuration module umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5 module umami.tests.unit.preprocessing.test_preprocessing_tools_resampling module umami.tests.unit.preprocessing.test_preprocessing_tools_scaling module umami.tests.unit.tf_tools module umami.tests.unit.tf_tools.test_generators module umami.tests.unit.tf_tools.test_tddgenerators module umami.tests.unit.tf_tools.test_tf_tools module umami.tests.unit.tf_tools.test_tf_tools_convert_to_record module umami.tests.unit.tools module umami.tests.unit.tools.test_tools module umami.tests.unit.train_tools module umami.tests.unit.train_tools.test_nn_tools module umami.tf_tools module umami.tf_tools.convert_to_record module umami.tf_tools.generators module umami.tf_tools.layers module umami.tf_tools.load_tfrecord module umami.tf_tools.models module umami.tf_tools.tddgenerators module umami.tf_tools.tools module umami.tools module umami.tools.tools module umami.tools.yaml_tools module umami.train module umami.train_tools module umami.train_tools.configuration module umami.train_tools.nn_tools module UmamiConditionGenerator (class in umami.tf_tools.generators) UmamiGenerator (class in umami.tf_tools.generators) UnderSampling (class in umami.preprocessing_tools.resampling.count_sampling) UnderSamplingNoReplace (class in umami.preprocessing_tools.resampling.importance_sampling_no_replace) UnderSamplingNoReplaceTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_resampling) UnderSamplingTestCase (class in umami.tests.unit.preprocessing.test_preprocessing_tools_resampling) UnderSamplingTestCaseEasyDistr (class in umami.tests.unit.preprocessing.test_preprocessing_tools_resampling) use_atlas_tag (umami.preprocessing_tools.configuration.GeneralSettings attribute) (umami.train_tools.configuration.ValidationSettingsConfig attribute) use_sample_weights (umami.train_tools.configuration.NNStructureConfig attribute) use_validation_samples (umami.preprocessing_tools.configuration.Sampling attribute)

V

val_batch_size (umami.train_tools.configuration.ValidationSettingsConfig attribute) validation_files (umami.train_tools.configuration.TrainConfigurationObject attribute)

ValidationSettingsConfig (class in umami.train_tools.configuration) var_file (umami.preprocessing_tools.configuration.GeneralSettings attribute)

W

Weighting (class in umami.preprocessing_tools.resampling.weighting) weighting_target_flavour (umami.preprocessing_tools.configuration.SamplingOptions attribute) working_point (umami.train_tools.configuration.EvaluationSettingsConfig attribute) (umami.train_tools.configuration.ValidationSettingsConfig attribute)

write_file() (umami.preprocessing_tools.resampling.resampling_base.Resampling method) (umami.preprocessing_tools.ttbar_merge.TTbarMerge method) write_tfrecord() (umami.tf_tools.convert_to_record.H5ToTFRecords method) write_train_sample() (umami.preprocessing_tools.writing_train_file.TrainSampleWriter method)

X

x_axis_granularity (umami.train_tools.configuration.EvaluationSettingsConfig attribute)

Y

YAML (class in umami.tools.yaml_tools)

yellow (umami.configuration.configuration.CustomFormatter attribute)