xgb.py 文件源码

def try_params( n_iterations, params, get_predictions = False ):

    n_estimators = int( round( n_iterations * trees_per_iteration ))
    print "n_estimators:", n_estimators
    pprint( params )

    model = XGB( n_estimators = n_estimators, nthread = -1, **params )

    return train_and_eval_sklearn_regressor( model, data )