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

Fixed bug in eta decay (+2 squashed commits)

Browse Source 
Squashed commits:
[b67caf2] Fix build
[365ceaa] Fixed bug in eta decay
This commit is contained in:
terrytangyuan
2015-10-31 12:40:19 -04:00
parent 888edba03f
commit 15a0d27eed
3 changed files with 1 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/python/test_models.py
View File

@@ -28,9 +28,6 @@ class TestModels(unittest.TestCase):
# learning_rates as a list
bst = xgb.train(param, dtrain, num_round, watchlist, learning_rates=[0.4, 0.3])
assert isinstance(bst, xgb.core.Booster)
# different length
num_round = 4
self.assertRaises(ValueError, xgb.train, param, dtrain, num_round, watchlist, learning_rates=[0.4, 0.3, 0.2])
# learning_rates as a customized decay function
def eta_decay(ithround, num_boost_round):