Python pickle 模型导入导出

训练好的模型想导出下次直接导入使用，就可

以用pickle

这里把训练好的model 以及tdif都保存到本地了

import pickle
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.naive_bayes import MultinomialNB

tfidf= TfidfVectorizer()
x_train_tfidf = tfidf.fit_transform(X_train)
x_test_tfidf = tfidf.transform(processContent)
with open('./xxxxx.pickle', 'wb') as f:
    tfidf=pickle.dump(tfidf, f)  
 
mnb = MultinomialNB()
mnb.fit(x_train_cv.toarray(), y_train)
with open('./xxxxxx.pickle', 'wb') as f:
    pickle.dump(mnb, f) 

这里就是导入 

with open('./logTfidf.pickle', 'rb') as f:
    cv=pickle.load(f) 

with open('./tfidflogModel.pickle', 'rb') as f:
    mnb = pickle.load(f)