How to save classifier to disk in scikit-learn

Question

Hi Guys,

I have trained the Naive Bayes classifier. I want to save this model for future use. How do I save a trained Naive Bayes classifier to disk and use it to predict data?

Answer 1

Hi@akhtar,

Classifiers are just objects that can be pickled and dumped like any other. You can go through the below code for your reference.

import cPickle
# save the classifier
with open('my_dumped_classifier.pkl', 'wb') as fid:
    cPickle.dump(gnb, fid)    
# load it again
with open('my_dumped_classifier.pkl', 'rb') as fid:
    gnb_loaded = cPickle.load(fid)