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

TST: Added test for show_stdv when using cv

Browse Source
This commit is contained in:
terrytangyuan 2015-10-04 22:42:39 -05:00
parent 0c360fe55f
commit dfb89e3442
1 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tests/python/test_models.py
View File

@ -46,3 +46,26 @@ def test_fpreproc():
return (dtrain, dtest, param) return (dtrain, dtest, param)
xgb.cv(param, dtrain, num_round, nfold=5, xgb.cv(param, dtrain, num_round, nfold=5,
metrics={'auc'}, seed = 0, fpreproc = fpreproc) metrics={'auc'}, seed = 0, fpreproc = fpreproc)
def test_show_stdv():
param = {'max_depth':2, 'eta':1, 'silent':1, 'objective':'binary:logistic'}
num_round = 2
xgb.cv(param, dtrain, num_round, nfold=5,
metrics={'error'}, seed = 0, show_stdv = False)