Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
num_round = 20
param = {'booster' : 'gbtree' }
param['tree_method'] = 'gpu_hist'
start_time = time.perf_counter()
bst = xgb.train(param,train,num_round)
end_time = time.perf_counter()
print('This took {0:.3f} seconds'.format(end_time-start_time))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment