DL1r-PFlow-Training-config.yaml 2.74 KB
Newer Older
1
# Set modelname and path to Pflow preprocessing config file
2
3
model_name: DL1r-PFlow_new-taggers-stats-22M
preprocess_config: examples/PFlow-Preprocessing.yaml
4

Alexander Froch's avatar
Alexander Froch committed
5
6
7
8
# Add here a pretrained model to start with.
# Leave empty for a fresh start
model_file:

9
# Add training file
Binbin Dong's avatar
Binbin Dong committed
10
train_file: /eos/user/b/bdong/DUQ/UmamiTrain/preprocessed/PFlow-hybrid-preprocessed_shuffled.h5
11
12
13

# Add validation files
# ttbar val
Binbin Dong's avatar
Binbin Dong committed
14
validation_file: /eos/user/b/bdong/DUQ/UmamiTrain/hybrids/MC16d_hybrid_odd_100_PFlow-no_pTcuts-file_0.h5
15

16
# zprime val
Binbin Dong's avatar
Binbin Dong committed
17
add_validation_file: /eos/user/b/bdong/DUQ/UmamiTrain/hybrids/MC16d_hybrid-ext_odd_0_PFlow-no_pTcuts-file_0.h5
18

19
20
ttbar_test_files:
    ttbar_r21:
Binbin Dong's avatar
Binbin Dong committed
21
        Path: /eos/user/b/bdong/DUQ/UmamiTrain/hybrids/MC16d_hybrid_odd_100_PFlow-no_pTcuts-file_1.h5
22
        data_set_name: "ttbar"
23

24
zpext_test_files:
Manuel Guth's avatar
Manuel Guth committed
25
    zpext_r21:
Binbin Dong's avatar
Binbin Dong committed
26
        Path: /eos/user/b/bdong/DUQ/UmamiTrain//hybrids/MC16d_hybrid-ext_odd_0_PFlow-no_pTcuts-file_1.h5
27
        data_set_name: "zpext"
28
29

# Path to Variable dict used in preprocessing
30
var_dict: umami/configs/DL1r_Variables.yaml
31
32
33

bool_use_taus: False

34
35
exclude: []

Manuel Guth's avatar
Manuel Guth committed
36
NN_structure:
Binbin Dong's avatar
Binbin Dong committed
37
    lr: 0.005
38
39
    batch_size: 15000
    activations: ["relu", "relu", "relu", "relu", "relu", "relu", "relu", "relu"]
Manuel Guth's avatar
Manuel Guth committed
40
    units: [256, 128, 60, 48, 36, 24, 12, 6]
Binbin Dong's avatar
Binbin Dong committed
41
    dropout_rate: [0.1, 0.2, 0.2, 0.2, 0.2, 0.2, 0.2, 0.2]
42

43
44
45
46
47
# Eval parameters for validation evaluation while training
Eval_parameters_validation:
    # Number of jets used for validation
    n_jets: 3e5

48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
    # Define taggers that are used for comparison in evaluate_model
    # This can be a list or a string for only one tagger
    tagger: ["rnnip", "DL1r"]

    # Define fc values for the taggers
    fc_values_comp: {
        "rnnip": 0.08,
        "DL1r": 0.018,
    }

    # Define fb values for the taggers
    fb_values_comp: {
        "rnnip": 0.2,
        "DL1r": 0.2,
    }

    # Charm fraction value used for evaluation of the trained model
65
66
    fc_value: 0.018

67
68
69
70
71
72
73
74
75
    # B fraction value used for evaluation
    fb_value: 0.2

    # Tau fraction value used for evaluation (for c)
    ftauforc_value: None

    # # Tau fraction value used for evaluation (for b)
    ftauforb_value: None

76
77
    # b Working point used in the evaluation
    WP_b: 0.77
78
79
80
81
82
83
84
85
86
87
    # C Working point used in the evaluation
    WP_c: 0.4

    # Set minimum of accuracy plot y-axis
    acc_ymin: 0.59
    # Set maximum of accuracy plot y-axis
    acc_ymax: 1.0

    # A list to add available variables to the evaluation files
    add_variables_eval: ["actualInteractionsPerCrossing"]
88
89
90
91
92
93
94
95
96

    # Enable/Disable atlas tag
    UseAtlasTag: True

    # fc_value and WP_b are autmoatically added to the plot label
    AtlasTag: "Internal Simulation"
    SecondTag: "\n$\\sqrt{s}=13$ TeV, PFlow jets"

    # Set the datatype of the plots
97
    plot_datatype: "pdf"