如何在Python中保存類的實例

Question

有沒有辦法在Python中的內存或文件系統中保存類的實例？我可以用shelve來做到這一點嗎？

以下行是this tutorial的一部分，需要很長時間才能執行，我需要將其緩存以供下一個程序執行。

clf = MultinomialNB().fit(X_train_counts, training_data['targets']) 

的clf類型：

>>> type(clf) 
<class 'sklearn.naive_bayes.MultinomialNB'> 

Answer 1

是的，你可以使用shelve堅持一個類的實例。 shelve爲您提供了一個字典界面，使過程相對透明。

下面，shelve使用pickle library;如果shelve API不符合您的需求，您可以直接進入該模塊。

scikit-learn明確支持pickle，看到Model persistence：

After training a scikit-learn model, it is desirable to have a way to persist the model for future use without having to retrain. The following section gives you an example of how to persist a model with pickle.