Revert "Merge branch 'wip-add-onnx-json' into 'main'"
Compare changes
Files
7+ 1
− 85
{"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]}
\ No newline at end of file