Skip to content

GitLab

Unverified Commit 55e9446a authored by Nikita Titov's avatar Nikita Titov Committed by GitHub
Browse files

[python] improve warning message about aliases in `cv()` function (#4766)

parent 2394b41a
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python-package/lightgbm/engine.py
View file @ 55e9446a
...@@ -570,7 +570,7 @@ def cv(params, train_set, num_boost_round=100, ...@@ -570,7 +570,7 @@ def cv(params, train_set, num_boost_round=100,
params['objective'] = 'none' params['objective'] = 'none'
for alias in _ConfigAliases.get("num_iterations"): for alias in _ConfigAliases.get("num_iterations"):
if alias in params: if alias in params:
_log_warning(f"Found `{alias}` in params. Will use it instead of argument") _log_warning(f"Found '{alias}' in params. Will use it instead of 'num_boost_round' argument")
num_boost_round = params.pop(alias) num_boost_round = params.pop(alias)
params["num_iterations"] = num_boost_round params["num_iterations"] = num_boost_round
if early_stopping_rounds is not None and early_stopping_rounds > 0: if early_stopping_rounds is not None and early_stopping_rounds > 0:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment