cause this code test pickle the booster, so change bst2 -> bst3
This commit is contained in:
parent
cdbafafc04
commit
d52d0ee9ed
@ -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
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user