From b587dd27041198a178d772a39ecaba922f1278dc Mon Sep 17 00:00:00 2001 From: Takahisa Shimoda Date: Fri, 23 Oct 2015 05:37:13 +0900 Subject: [PATCH 1/2] fix training.py for evals_result in python3 --- python-package/xgboost/training.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python-package/xgboost/training.py b/python-package/xgboost/training.py index 1e7294d7b..ae12fd868 100644 --- a/python-package/xgboost/training.py +++ b/python-package/xgboost/training.py @@ -78,7 +78,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None, res = re.findall("([0-9a-zA-Z@]+[-]*):-?([0-9.]+).", msg) for key in evals_name: evals_idx = evals_name.index(key) - res_per_eval = len(res) / len(evals_name) + res_per_eval = len(res) // len(evals_name) for r in range(res_per_eval): res_item = res[(evals_idx*res_per_eval) + r] res_key = res_item[0] @@ -135,7 +135,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None, res = re.findall("([0-9a-zA-Z@]+[-]*):-?([0-9.]+).", msg) for key in evals_name: evals_idx = evals_name.index(key) - res_per_eval = len(res) / len(evals_name) + res_per_eval = len(res) // len(evals_name) for r in range(res_per_eval): res_item = res[(evals_idx*res_per_eval) + r] res_key = res_item[0] From 607599f2a1f6a50eda95ac1282492a8ecdc2913d Mon Sep 17 00:00:00 2001 From: Takahisa Shimoda Date: Fri, 23 Oct 2015 05:40:31 +0900 Subject: [PATCH 2/2] fix sklearn.py for evals_result in python3 --- python-package/xgboost/sklearn.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python-package/xgboost/sklearn.py b/python-package/xgboost/sklearn.py index 3bf747b58..abfae6b4a 100644 --- a/python-package/xgboost/sklearn.py +++ b/python-package/xgboost/sklearn.py @@ -190,7 +190,7 @@ class XGBModel(XGBModelBase): if evals_result: for val in evals_result.items(): - evals_result_key = val[1].keys()[0] + evals_result_key = list(val[1].keys())[0] evals_result[val[0]][evals_result_key] = val[1][evals_result_key] self.evals_result_ = evals_result @@ -341,7 +341,7 @@ class XGBClassifier(XGBModel, XGBClassifierBase): if evals_result: for val in evals_result.items(): - evals_result_key = val[1].keys()[0] + evals_result_key = list(val[1].keys())[0] evals_result[val[0]][evals_result_key] = val[1][evals_result_key] self.evals_result_ = evals_result