From e339cdec5222fa858a090a905a35ecf22b47e997 Mon Sep 17 00:00:00 2001 From: Johan Manders Date: Mon, 12 Oct 2015 16:47:24 +0200 Subject: [PATCH] Too many branches and unused key --- python-package/xgboost/sklearn.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/python-package/xgboost/sklearn.py b/python-package/xgboost/sklearn.py index b3d973928..3c279bdb6 100644 --- a/python-package/xgboost/sklearn.py +++ b/python-package/xgboost/sklearn.py @@ -190,8 +190,7 @@ class XGBModel(XGBModelBase): if eval_results: for val in eval_results.items(): - for k, v in val[1].items(): - eval_results[val[0]] = np.array(v, dtype=float) + eval_results[val[0]] = [np.array(v[1], dtype=float) for v in val[1].items()] self.eval_results = eval_results if early_stopping_rounds is not None: @@ -305,8 +304,7 @@ class XGBClassifier(XGBModel, XGBClassifierBase): if eval_results: for val in eval_results.items(): - for k, v in val[1].items(): - eval_results[val[0]] = np.array(v, dtype=float) + eval_results[val[0]] = [np.array(v[1], dtype=float) for v in val[1].items()] self.eval_results = eval_results if early_stopping_rounds is not None: