Update main_glm_gbm.py