Save classifier to disk in scikit-learn

by

Classifiers are just objects that can be pickled and dumped like any other. To continue your example:

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)

Edit: if you are using a sklearn Pipeline in which you have custom transformers that cannot be serialized by pickle (nor by joblib), then using Neuraxle’s custom ML Pipeline saving is a solution where you can define your own custom step savers on a per-step basis. The savers are called for each step if defined upon saving, and otherwise joblib is used as default for steps without a saver.

More Related Contents:

  1. How to get most informative features for scikit-learn classifiers?
  2. Scikit-learn: How to obtain True Positive, True Negative, False Positive and False Negative
  3. Mixing categorial and continuous data in Naive Bayes classifier using scikit-learn
  4. scikit-learn .predict() default threshold
  5. How to extract the decision rules from scikit-learn decision-tree?
  6. scikit-learn & statsmodels – which R-squared is correct?
  7. Predict classes or class probabilities?
  8. Why is Random Forest with a single tree much better than a Decision Tree classifier?
  9. Feature/Variable importance after a PCA analysis
  10. Recovering features names of explained_variance_ratio_ in PCA with sklearn
  11. LabelEncoder for categorical features?
  12. What is exactly sklearn.pipeline.Pipeline?
  13. difference between StratifiedKFold and StratifiedShuffleSplit in sklearn
  14. Scikit Learn OneHotEncoder fit and transform Error: ValueError: X has different shape than during fitting
  15. memory issues when transforming np.array using to_categorical
  16. Can anyone explain me StandardScaler?
  17. How to find the importance of the features for a logistic regression model?
  18. TfidfVectorizer in scikit-learn : ValueError: np.nan is an invalid document
  19. scikit-learn random state in splitting dataset
  20. Scikit Learn GridSearchCV without cross validation (unsupervised learning)
  21. How to compute precision, recall, accuracy and f1-score for the multiclass case with scikit learn?
  22. Save Naive Bayes Trained Classifier in NLTK
  23. Tensorflow Precision / Recall / F1 score and Confusion matrix
  24. Using Smote with Gridsearchcv in Scikit-learn
  25. confused about random_state in decision tree of scikit learn
  26. Does GridSearchCV perform cross-validation?
  27. Can I send callbacks to a KerasClassifier?
  28. How to get most informative features for scikit-learn classifier for different class?
  29. Model help using Scikit-learn when using GridSearch
  30. Determining the most contributing features for SVM classifier in sklearn

Leave a Comment