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

This commit is contained in:
王煜杰 2016-02-25 14:49:33 +08:00
parent cdbafafc04
commit d52d0ee9ed

View File

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