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

Too many branches and unused key

Browse Source
This commit is contained in:
Johan Manders 2015-10-12 16:47:24 +02:00
parent 40566cdbba
commit e339cdec52
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
python-package/xgboost/sklearn.py
View File

@ -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: