保存分类器以供将来预测

时间:2018-03-04 18:22:13

标签: python machine-learning scikit-learn pickle sentiment-analysis

我是使用python进行机器学习的新手,我正在尝试构建一个使用此datasettutorial的情感分析器。一切都在测试数据上运行良好。但我正在尝试保存我的分类器以备将来使用。我正在使用泡菜将其保存为

sentiment_analyzer = open("Sentiment_Analyzer.pkl", "wb")
pkl.dump(classifier_linear, sentiment_analyzer)
sentiment_analyzer.close()

后来,我正在通过这个来提取我保存的分析器

model_pkl = open("Sentiment_Analyzer.pkl", "rb")
model = pkl.load(model_pkl)

但我无法理解如何在我提取的predict分类器上调用model方法。

1 个答案:

答案 0 :(得分:0)

您也需要保存vectorizer,就像腌制模型一样。然后在将来使用期间,加载矢量化器和分类器,使用加载的矢量化器转换新的X,然后在分类器上调用predict()