Commit fa21fd57 authored by Binbin Dong's avatar Binbin Dong
Browse files

added Dropout in DL1 model

parent 9fbbdbe3
Pipeline #2936867 passed with stages
in 18 minutes and 12 seconds
...@@ -7,37 +7,25 @@ preprocess_config: examples/PFlow-Preprocessing.yaml ...@@ -7,37 +7,25 @@ preprocess_config: examples/PFlow-Preprocessing.yaml
model_file: model_file:
# Add training file # Add training file
train_file: /work/ws/nemo/fr_af1100-Training-Simulations-0/preprocessed/PFlow-hybrid-preprocessed_shuffled.h5 train_file: /eos/user/b/bdong/DUQ/UmamiTrain/preprocessed/PFlow-hybrid-preprocessed_shuffled.h5
# Add validation files # Add validation files
# ttbar val # ttbar val
validation_file: /work/ws/nemo/fr_af1100-Training-Simulations-0/hybrids/MC16d_hybrid_odd_100_PFlow-no_pTcuts-file_0.h5 validation_file: /eos/user/b/bdong/DUQ/UmamiTrain/hybrids/MC16d_hybrid_odd_100_PFlow-no_pTcuts-file_0.h5
# zprime val # zprime val
add_validation_file: /work/ws/nemo/fr_af1100-Training-Simulations-0/hybrids/MC16d_hybrid-ext_odd_0_PFlow-no_pTcuts-file_0.h5 add_validation_file: /eos/user/b/bdong/DUQ/UmamiTrain/hybrids/MC16d_hybrid-ext_odd_0_PFlow-no_pTcuts-file_0.h5
ttbar_test_files: ttbar_test_files:
ttbar_r21: ttbar_r21:
Path: /work/ws/nemo/fr_af1100-Training-Simulations-0/hybrids/MC16d_hybrid_odd_100_PFlow-no_pTcuts-file_1.h5 Path: /eos/user/b/bdong/DUQ/UmamiTrain/hybrids/MC16d_hybrid_odd_100_PFlow-no_pTcuts-file_1.h5
data_set_name: "ttbar" data_set_name: "ttbar"
ttbar_r22:
Path: /work/ws/nemo/fr_af1100-Training-Simulations-0/hybrids_r22/MC16d_hybrid-r22_odd_100_PFlow-no_pTcuts-file_1.h5
data_set_name: "ttbar_comparison"
zpext_test_files: zpext_test_files:
zpext_r21: zpext_r21:
Path: /work/ws/nemo/fr_af1100-Training-Simulations-0/hybrids/MC16d_hybrid-ext_odd_0_PFlow-no_pTcuts-file_1.h5 Path: /eos/user/b/bdong/DUQ/UmamiTrain//hybrids/MC16d_hybrid-ext_odd_0_PFlow-no_pTcuts-file_1.h5
data_set_name: "zpext" data_set_name: "zpext"
zpext_r22:
Path: /work/ws/nemo/fr_af1100-Training-Simulations-0/hybrids_r22/MC16d_hybrid-r22-ext_odd_0_PFlow-no_pTcuts-file_1.h5
data_set_name: "zpext_comparison"
zpext_r22_no_QSP:
Path: /work/ws/nemo/fr_af1100-Training-Simulations-0/hybrids_r22/MC16d_hybrid-r22-ext_odd_0_PFlow-no_pTcuts_No_QSPI-file_1.h5
data_set_name: "zpext_comparison_no_QSP"
# Path to Variable dict used in preprocessing # Path to Variable dict used in preprocessing
var_dict: umami/configs/DL1r_Variables.yaml var_dict: umami/configs/DL1r_Variables.yaml
...@@ -46,10 +34,11 @@ bool_use_taus: False ...@@ -46,10 +34,11 @@ bool_use_taus: False
exclude: [] exclude: []
NN_structure: NN_structure:
lr: 0.01 lr: 0.005
batch_size: 15000 batch_size: 15000
activations: ["relu", "relu", "relu", "relu", "relu", "relu", "relu", "relu"] activations: ["relu", "relu", "relu", "relu", "relu", "relu", "relu", "relu"]
units: [256, 128, 60, 48, 36, 24, 12, 6] units: [256, 128, 60, 48, 36, 24, 12, 6]
dropout_rate: [0.1, 0.2, 0.2, 0.2, 0.2, 0.2, 0.2, 0.2]
# Eval parameters for validation evaluation while training # Eval parameters for validation evaluation while training
Eval_parameters_validation: Eval_parameters_validation:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment