pickle用于在训练模型后将模型保存下来,避免重复运行,可以节省时间,还可以把训练好的模型直接导入其他文件
模型保存
tfidf_path = 'tfidf.pkl'
with open(tfidf_path, 'wb') as in_data:
pickle.dump(clf_tfidf, in_data, pickle.HIGHEST_PROTOCOL)
print("tfidf model saved:" + tfidf_path)
wb: 以二进制格式打开一个文件只用于写入。如果该文件已存在则将其覆盖。如果该文件不存在,创建新文件。
pickle.dump(obj, file, [,protocol]