使用sklearn训练好的模型和CountVectorizer的保存以及模型调用
1.概述
2.模型的保存
3.模型的调用
1.概述
对于已经训练好的模型是需要进行保存操作饿,否则每一次的使用都会重新再次训练,而模型的执行效率堪忧。为此本文利用joblib和pickle分别对分类模型进行磁盘保存,生成model.pkl和feature.pkl文件,在调用的过程中直接加载这两个文件就ok了。
2.模型的保存
下面展示一些 分类模型训练代码
。
对于分类模型可以参考上一篇文章,此处仅展示部分函数。
// An highlighted block
model,vec = classifier_model(x_train,y_train)
#对分类模型保存
joblib.dump(model