Skip to content
Snippets Groups Projects

Revert "Merge branch 'wip-add-onnx-json' into 'main'"

Merged Tobiasz Fic requested to merge revert-0cff23e1 into main
7 files
+ 5
239
Compare changes
  • Side-by-side
  • Inline
Files
7
{
{"file_names": {"training": "/lustre/cbm/users/tfic/pid_plain_trees/trees/PlainTree50K_DCM_trdrichrec_12agev.root", "test": "/lustre/cbm/users/tfic/pid_plain_trees/trees/PlainTree5K_DCM_trdrichrec_12agev.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.38661597183346746, 1.4825726561740278, 2.0401938866642655, 2.7015859877268475, 3.7153128963006865, 12.0]}
"file_names": {
\ No newline at end of file
"training": "/lustre/cbm/users/tfic/pid_plain_trees/trees/PlainTree50K_DCM_trdrichrec_12agev.root",
"test": "/lustre/cbm/users/tfic/pid_plain_trees/trees/PlainTree5K_DCM_trdrichrec_12agev.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_vtx_chi2",
"Complex_chi2_ov_ndf_vtx",
"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.38661597183346746,
1.4825726561740278,
2.0401938866642655,
2.7015859877268475,
3.7153128963006865,
12.0
]
}
\ No newline at end of file
Loading