Search code examples
python-3.xscikit-learn

Which estimators in scikit-learn support `partial_fit` API?


Only examples of estimators that implement the partial_fit API that I can find are SGDRegressor and SGDClassifier. Are there others? Is there a list of estimators that implement the partial_fit API?


Solution

  • Here is the list of models that allows partial_fit().

    Reference

    Classification

    sklearn.naive_bayes.MultinomialNB
    sklearn.naive_bayes.BernoulliNB sklearn.linear_model.Perceptron
    sklearn.linear_model.SGDClassifier
    sklearn.linear_model.PassiveAggressiveClassifier

    Regression

    sklearn.linear_model.SGDRegressor
    sklearn.linear_model.PassiveAggressiveRegressor

    Clustering

    sklearn.cluster.MiniBatchKMeans

    Decomposition / feature Extraction

    sklearn.decomposition.MiniBatchDictionaryLearning
    sklearn.cluster.MiniBatchKMeans