test.py 文件源码

def tune_num_boost_round():
    # global watchlist
    global num_boost_round
    global  evals_result
    evals_result = {}
    xgb.train(params=params_no_sklearn,dtrain=dtrain,num_boost_round=num_boost_round,evals=watchlist,evals_result=evals_result)
    evals_result = evals_result['eval']['map']
    pprint.pprint(evals_result)
    max = 0.0
    max_loc = 0
    for i,v in enumerate(evals_result):
        # print '%d ...... %d : %d'%(i,max_loc,max)
        if v>max:
            max = v
            max_loc = i
    print "max_loc : %d ,  max : %d"%(max_loc,max)
    num_boost_round = max_loc+1
    print '****  num_boost_round : ', num_boost_round