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

XGBModel.fit had a call to DMatrix without missing=self.missing. fixed that

Browse Source
This commit is contained in:
borundev 2016-05-07 12:32:03 +02:00
parent 6e79ba831a
commit 95bcff90af
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/sklearn.py
View File

@ -222,7 +222,7 @@ class XGBModel(XGBModelBase):
evals_result = {}
if eval_set is not None:
evals = list(DMatrix(x[0], label=x[1]) for x in eval_set)
evals = list(DMatrix(x[0], label=x[1],missing=self.missing) for x in eval_set)
evals = list(zip(evals, ["validation_{}".format(i) for i in
range(len(evals))]))
else: