model.py 文件源码

def run_gbm(dtrain, dvalid, param):
    # check training arguments in param
    n_round = param.get('num_boost_round', 5000)
    early_stop = param.get('early_stopping_rounds', 5)
    verbose_eval = param.get('verbose_eval', 100)
    # specify validations set to watch performance
    watchlist = [(dvalid, 'eval')]
    bst = xgb.train(params=param,
                    dtrain=dtrain,
                    num_boost_round=n_round,
                    evals=watchlist,
                    early_stopping_rounds=early_stop,
                    verbose_eval = verbose_eval
                   )
    return bst