Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

make travis happy

Browse Source
This commit is contained in:
Vadim Khotilovich 2016-06-04 22:38:24 -05:00
parent ba04a1d552
commit 26a82621a2
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/python/test_basic_models.py
View File

@ -111,6 +111,7 @@ class TestModels(unittest.TestCase):
param['eval_metric'] = ["auc", "logloss", 'error'] param['eval_metric'] = ["auc", "logloss", 'error']
evals_result = {} evals_result = {}
bst = xgb.train(param, dtrain, 4, watchlist, evals_result=evals_result) bst = xgb.train(param, dtrain, 4, watchlist, evals_result=evals_result)
assert isinstance(bst, xgb.core.Booster)
assert len(evals_result['eval']) == 3 assert len(evals_result['eval']) == 3
assert set(evals_result['eval'].keys()) == {'auc', 'error', 'logloss'} assert set(evals_result['eval'].keys()) == {'auc', 'error', 'logloss'}