保存分類器以供將來預測

Question

我是python機器學習的新手，我正在嘗試構建一個Sentiment Analyzer，在其中使用此數據集和本教程 。 測試數據一切正常。 但是我正在嘗試保存我的分類器以備將來使用。 我正在使用泡菜通過將其保存為

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() 。