Skip to content
This repository has been archived by the owner on Jul 27, 2018. It is now read-only.

How to save and restore model #33

Open
Shuang0420 opened this issue Feb 23, 2018 · 0 comments
Open

How to save and restore model #33

Shuang0420 opened this issue Feb 23, 2018 · 0 comments

Comments

@Shuang0420
Copy link

Shuang0420 commented Feb 23, 2018

I tried to save and load model as I usually do when using sklearn model, but the evaluation result of model is apparently not the same. Anything wrong with my code?

save model

tree = DecisionTreeClassifier(max_depth=1, min_samples_leaf=1)
bag = Bagging(base_classifier=tree, n_classifiers=10)
bag.fit(text, labels)
with open(bag_path, "wb") as f:
  pickle.dump(bag, f)

load model

with open(self.bag_path, "rb") as f:
  self.bag = pickle.load(f)
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant