保存分类器以供将来预测

Question

我是使用 python 进行机器学习的新手，我正在尝试构建一个情感分析器，我在其中使用这个dataset 和这个tutorial。测试数据一切正常。但我正在尝试保存我的分类器以备将来使用。我正在使用 pickle 将其保存为

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)

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

Answer 1

您还需要保存vectorizer，就像您腌制模型一样。然后在以后的使用中，加载向量器和分类器，使用加载的向量器转换新的 X，然后在分类器上调用 predict()。