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

[py] eta decay bugfix

Browse Source
This commit is contained in:
Faron
2016-04-30 12:32:49 +02:00
parent 9bc2ac4bd0
commit ad3f49e881
2 changed files with 33 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -108,6 +108,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None,
bst = Booster(params, [dtrain] + [d[0] for d in evals])
_params = dict(params) if isinstance(params, list) else params
_eta_param_name = 'eta' if 'eta' in _params else 'learning_rate'
if 'num_parallel_tree' in _params:
num_parallel_tree = _params['num_parallel_tree']
nboost //= num_parallel_tree
@@ -168,9 +169,9 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None,
for i in range(start_iteration, num_boost_round):
if learning_rates is not None:
if isinstance(learning_rates, list):
bst.set_param({'eta': learning_rates[i]})
bst.set_param(_eta_param_name, learning_rates[i])
else:
bst.set_param({'eta': learning_rates(i, num_boost_round)})
bst.set_param(_eta_param_name, learning_rates(i, num_boost_round))
# Distributed code: need to resume to this point.
# Skip the first update if it is a recovery step.