Train a different BDT with hep_ml

umd-lhcb / MuonBDTPid

Muon PID with a uboost BDT (in ROOT 5). Also include code for PID efficiency studies

0 stars 0 forks source link

Train a different BDT with hep_ml #2

Closed yipengsun closed 3 years ago

yipengsun commented 4 years ago

Greg suggested that we may want to retrain a UBosst BDT with hep_ml. Using Python has many upsides, but so far it seems that the only ways to export a trained BDT is either use Python's built-in pickle or joblib. I'm not sure about the cross-Python-version compatibility of the exported.

yipengsun commented 4 years ago

Well, the good news is that pickle is guaranteed to be backward-compatible: https://docs.python.org/3/library/pickle.html#relationship-to-other-python-modules

yipengsun commented 3 years ago

Greg shared the ntuples he used to train the OG BDT. I'll start working on it now.

yipengsun commented 3 years ago

Probably we'll not retrain a new BDT unless the existing UBDT misbehaves. Closed for now.