Adopted from XGBoost Benchmarking examples here: https://github.com/RAMitchell/GBM-Benchmarks