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

GLM test unit: make run deterministic (#2147)

Browse Source
This commit is contained in:
Laurae 2017-03-24 16:54:39 +01:00 committed by Tianqi Chen
parent f1fe024a9d
commit 5c13aa0a8a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/python/test_basic_models.py
View File

@ -12,7 +12,7 @@ rng = np.random.RandomState(1994)
class TestModels(unittest.TestCase): class TestModels(unittest.TestCase):
def test_glm(self): def test_glm(self):
param = {'silent': 1, 'objective': 'binary:logistic', param = {'silent': 1, 'objective': 'binary:logistic',
'booster': 'gblinear', 'alpha': 0.0001, 'lambda': 1} 'booster': 'gblinear', 'alpha': 0.0001, 'lambda': 1, 'nthread': 1}
watchlist = [(dtest, 'eval'), (dtrain, 'train')] watchlist = [(dtest, 'eval'), (dtrain, 'train')]
num_round = 4 num_round = 4
bst = xgb.train(param, dtrain, num_round, watchlist) bst = xgb.train(param, dtrain, num_round, watchlist)