Skip to content

MVAUtils: Adding support for XGBoost

Yuan-Tang Chou requested to merge yuchou/athena:21.2-XGBoost-MVAUtils into 21.2

Adding derived class to support XGboost node/forest in MVAUtils. A python utility to convert the model saved by XGBoost training to our TTree format is also provided. Implementation follows closely with what was done in lgbm, presently only support regression and binary classification.

Merge request reports