Skip to content

Instantly share code, notes, and snippets.

@zronaghi

zronaghi/gc-xgb-train.py

Created March 8, 2021 02:42
Show Gist options
  • Save zronaghi/6960a5b8814c39b232751656d1ea6c3f to your computer and use it in GitHub Desktop.
Save zronaghi/6960a5b8814c39b232751656d1ea6c3f to your computer and use it in GitHub Desktop.
Download ZIP
Raw
gc-xgb-train.py
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