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

update sklearn.py because evals_result in training.py changed

Browse Source 
Because I changed the training.py, the sklearn.py had to be changed also to be able to read all the data form evals_result.
This commit is contained in:
Johan Manders 2015-10-12 16:31:23 +02:00
parent b758a13813
commit 40566cdbba
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -187,10 +187,11 @@ class XGBModel(XGBModelBase):
early_stopping_rounds=early_stopping_rounds,
evals_result=eval_results, feval=feval,
verbose_eval=verbose)
if eval_results:
eval_results = {k: np.array(v, dtype=float)
for k, v in eval_results.items()}
eval_results = {k: np.array(v) for k, v in eval_results.items()}
for val in eval_results.items():
for k, v in val[1].items():
eval_results[val[0]] = np.array(v, dtype=float)
self.eval_results = eval_results
if early_stopping_rounds is not None:
@ -303,8 +304,9 @@ class XGBClassifier(XGBModel, XGBClassifierBase):
verbose_eval=verbose)
if eval_results:
eval_results = {k: np.array(v, dtype=float)
for k, v in eval_results.items()}
for val in eval_results.items():
for k, v in val[1].items():
eval_results[val[0]] = np.array(v, dtype=float)
self.eval_results = eval_results
if early_stopping_rounds is not None: