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

bugfix type-check xgb_model param

Browse Source
This commit is contained in:
Far0n 2015-11-03 14:43:08 +01:00
parent 4fe2f2fb09
commit b894f7c9d6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -63,7 +63,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None,
evals = list(evals) evals = list(evals)
ntrees = 0 ntrees = 0
if xgb_model is not None: 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() xgb_model = xgb_model.save_raw()
bst = Booster(params, [dtrain] + [d[0] for d in evals], model_file=xgb_model) bst = Booster(params, [dtrain] + [d[0] for d in evals], model_file=xgb_model)
ntrees = len(bst.get_dump()) ntrees = len(bst.get_dump())