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

2
python-package/xgboost/training.py
View File

@@ -123,7 +123,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None,
best_msg = ''
best_score_i = 0
if isinstance(learning_rates, list) and len(learning_rates) < num_boost_round:
if isinstance(learning_rates, list) and len(learning_rates) != num_boost_round:
raise ValueError("Length of list 'learning_rates' has to equal 'num_boost_round'.")
for i in range(num_boost_round):