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

cause this code test pickle the booster, so change bst2 -> bst3

Browse Source
This commit is contained in:
王煜杰 2016-02-25 14:49:33 +08:00
parent cdbafafc04
commit d52d0ee9ed
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
demo/guide-python/basic_walkthrough.py
View File

@ -42,7 +42,7 @@ assert np.sum(np.abs(preds2-preds)) == 0
pks = pickle.dumps(bst2) pks = pickle.dumps(bst2)
# load model and data in # load model and data in
bst3 = pickle.loads(pks) bst3 = pickle.loads(pks)
preds3 = bst2.predict(dtest2) preds3 = bst3.predict(dtest2)
# assert they are the same # assert they are the same
assert np.sum(np.abs(preds3-preds)) == 0 assert np.sum(np.abs(preds3-preds)) == 0