Fix a small typo in sklearn.py that broke multiple eval metrics (#5341)

2020-02-22 03:02:37 -08:00
parent 74e2f652de
commit 7d178cbd25
2 changed files with 3 additions and 2 deletions
--- a/python-package/xgboost/training.py
+++ b/python-package/xgboost/training.py
@@ -38,7 +38,7 @@ def _train_internal(params, dtrain,

    _params = dict(params) if isinstance(params, list) else params

-    if 'num_parallel_tree' in _params and params[
+    if 'num_parallel_tree' in _params and _params[
            'num_parallel_tree'] is not None:
        num_parallel_tree = _params['num_parallel_tree']
        nboost //= num_parallel_tree