Skip to content
Snippets Groups Projects

Fix slurm execution, add temoporary debug pipeline script for quick testing...

Merged Tobiasz Fic requested to merge hotfix-fix-batch into main
16 files
+ 167
290
Compare changes
  • Side-by-side
  • Inline
Files
16
+ 0
96
{
"file_names": {
"training": "/lustre/cbm/users/tfic/pid_plain_trees/trees/train/analysis_plain_ttree.root",
"test": "/lustre/cbm/users/tfic/pid_plain_trees/trees/validate/analysis_plain_ttree.root"
},
"var_names": {
"momentum": "Complex_p",
"charge": "Complex_q",
"mass2": "Complex_mass2",
"pid": "Complex_pid"
},
"vars_to_draw": [
"Complex_mass2",
"Complex_p"
],
"features_for_train": [
"Complex_pT",
"Complex_mass2",
"Complex_n_hits_trd",
"Complex_e_loss_0",
"Complex_e_loss_1",
"Complex_e_loss_2",
"Complex_e_loss_3",
"Complex_vtx_chi2",
"Complex_chi2_ov_ndf_vtx",
"Complex_chi2_ov_ndf_trd",
"Complex_M"
],
"cuts": {
"Complex_mass2": {
"lower": -1.0,
"upper": 2.0
},
"Complex_pT": {
"lower": 0.0,
"upper": 2.0
},
"Complex_p": {
"lower": -12.0,
"upper": 12.0
},
"Complex_eta": {
"lower": 0.0,
"upper": 6.0
}
},
"hyper_params": {
"values": {
"n_estimators": 776,
"max_depth": 7,
"learning_rate": 0.08112714035375232,
"gamma": 9,
"subsample": 0.71582,
"alpha": 17
},
"ranges": {
"n_estimators": [
300,
1200
],
"max_depth": [
2,
8
],
"learning_rate": [
0.01,
0.1
],
"gamma": [
0,
10
],
"subsample": [
0.3,
0.9
],
"alpha": [
0,
20
]
}
},
"bins": [
0.013909902337240055,
1.1882912607191654,
1.4851481804511315,
1.7604143725972687,
2.0431516134833427,
2.3525949053929613,
2.70833153379191,
3.144796421193885,
3.7330316230304086,
4.691929802443221,
12.0
]
}
\ No newline at end of file
Loading