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

Migrate pylint check to Python 3 (#4381)

Browse Source 
* Migrate lint to Python 3

* Fix lint errors

* Use Miniconda3 to use Python 3.7

* Use latest pylint and astroid
This commit is contained in:
Philip Hyunsu Cho
2019-04-21 01:01:54 -07:00
committed by GitHub
parent 5e97de6a41
commit bbe0dbd7ec
11 changed files with 118 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
python-package/xgboost/training.py
View File

@@ -49,7 +49,7 @@ def _train_internal(params, dtrain,
# Distributed code: Load the checkpoint from rabit.
version = bst.load_rabit_checkpoint()
assert(rabit.get_world_size() != 1 or version == 0)
assert rabit.get_world_size() != 1 or version == 0
rank = rabit.get_rank()
start_iteration = int(version / 2)
nboost += start_iteration
@@ -75,12 +75,12 @@ def _train_internal(params, dtrain,
bst.save_rabit_checkpoint()
version += 1
assert(rabit.get_world_size() == 1 or version == rabit.version_number())
assert rabit.get_world_size() == 1 or version == rabit.version_number()
nboost += 1
evaluation_result_list = []
# check evaluation result.
if len(evals) != 0:
if evals:
bst_eval_set = bst.eval_set(evals, i, feval)
if isinstance(bst_eval_set, STRING_TYPES):
msg = bst_eval_set
@@ -402,7 +402,7 @@ def cv(params, dtrain, num_boost_round=10, nfold=3, stratified=False, folds=None
else:
params = dict((k, v) for k, v in params.items())
if len(metrics) == 0 and 'eval_metric' in params:
if (not metrics) and 'eval_metric' in params:
if isinstance(params['eval_metric'], list):
metrics = params['eval_metric']
else:
@@ -462,7 +462,7 @@ def cv(params, dtrain, num_boost_round=10, nfold=3, stratified=False, folds=None
rank=0,
evaluation_result_list=res))
except EarlyStopException as e:
for k in results.keys():
for k in results:
results[k] = results[k][:(e.best_iteration + 1)]
break
if as_pandas: