From b894f7c9d65aa59cbd0f910998923a31b40b6d7d Mon Sep 17 00:00:00 2001 From: Far0n Date: Tue, 3 Nov 2015 14:43:08 +0100 Subject: [PATCH] bugfix type-check xgb_model param --- python-package/xgboost/training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python-package/xgboost/training.py b/python-package/xgboost/training.py index af9d6541d..03e24bdba 100644 --- a/python-package/xgboost/training.py +++ b/python-package/xgboost/training.py @@ -63,7 +63,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None, evals = list(evals) ntrees = 0 if xgb_model is not None: - if xgb_model is not isinstance(xgb_model, STRING_TYPES): + if not isinstance(xgb_model, STRING_TYPES): xgb_model = xgb_model.save_raw() bst = Booster(params, [dtrain] + [d[0] for d in evals], model_file=xgb_model) ntrees = len(bst.get_dump())