Demo for accessing eval metrics in xgboost
This commit is contained in:
parent
67f3c687b8
commit
6e2bdcbbbc
29
demo/guide-python/evals_result.py
Normal file
29
demo/guide-python/evals_result.py
Normal file
@ -0,0 +1,29 @@
|
||||
import xgboost as xgb
|
||||
##
|
||||
# This script demonstrate how to access the eval metrics in xgboost
|
||||
##
|
||||
dtrain = xgb.DMatrix('../data/agaricus.txt.train', silent=True)
|
||||
dtest = xgb.DMatrix('../data/agaricus.txt.test', silent=True)
|
||||
|
||||
param = [('max_depth', 2), ('objective', 'binary:logistic'), ('eval_metric', 'logloss'), ('eval_metric', 'error')]
|
||||
|
||||
num_round = 2
|
||||
watchlist = [(dtest,'eval'), (dtrain,'train')]
|
||||
|
||||
evals_result = {}
|
||||
bst = xgb.train(param, dtrain, num_round, watchlist, evals_result=evals_result)
|
||||
|
||||
print('Access logloss metric directly from evals_result:')
|
||||
print(evals_result['eval']['logloss'])
|
||||
|
||||
print('')
|
||||
print('Access metrics through a loop:')
|
||||
for e_name, e_mtrs in evals_result.items():
|
||||
print('- {}'.format(e_name))
|
||||
for e_mtr_name, e_mtr_vals in e_mtrs.items():
|
||||
print(' - {}'.format(e_mtr_name))
|
||||
print(' - {}'.format(e_mtr_vals))
|
||||
|
||||
print('')
|
||||
print('Access complete dictionary:')
|
||||
print(evals_result)
|
||||
Loading…
x
Reference in New Issue
Block a user