在Python中将分类器保存为pickle文件的问题

Question

我尝试将多项式naivebayes分类器保存为.pkl文件，但遇到了一个错误。在

我试过了

import pickle

with open(r'C:\Users\User\Desktop\clf.pkl','rb') as f:
    pickle.dump(mnb,f) 
#mnb is the MultinomialNB classifier

我得到的错误是

^{pr2}$

我需要先在某个地方创建一个文件吗？在

Answer 1

如果要保存它，请按'wb'写入模式而不是'rb'读取模式打开它，如下所示：

with open(r'C:\Users\User\Desktop\clf.pkl','wb') as f:
    pickle.dump(mnb,f)

In write mode, it will automatically create a file if not present!

它会解决你的问题。希望，这有帮助。在