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

Merge pull request #867 from wyj2046/master

Browse Source 
cause this code test pickle the booster, so change bst2 -> bst3
This commit is contained in:
Tianqi Chen 2016-02-25 20:25:47 -08:00
commit 61d9edcaa4
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