sklearn.utils.testing.all_estimators

sklearn.utils.testing.all_estimators(include_meta_estimators=False, include_other=False, type_filter=None, include_dont_test=False)[source]

Get a list of all estimators from sklearn.

This function crawls the module and gets all classes that inherit from BaseEstimator. Classes that are defined in test-modules are not included. By default meta_estimators such as GridSearchCV are also not included.

Parameters:
include_meta_estimators : boolean, default=False

Whether to include meta-estimators that can be constructed using an estimator as their first argument. These are currently BaseEnsemble, OneVsOneClassifier, OutputCodeClassifier, OneVsRestClassifier, RFE, RFECV.

include_other : boolean, default=False

Wether to include meta-estimators that are somehow special and can not be default-constructed sensibly. These are currently Pipeline, FeatureUnion and GridSearchCV

include_dont_test : boolean, default=False

Whether to include “special” label estimator or test processors.

type_filter : string, list of string, or None, default=None

Which kind of estimators should be returned. If None, no filter is applied and all estimators are returned. Possible values are ‘classifier’, ‘regressor’, ‘cluster’ and ‘transformer’ to get estimators only of these specific types, or a list of these to get the estimators that fit at least one of the types.

Returns:
estimators : list of tuples

List of (name, class), where name is the class name as string and class is the actuall type of the class.