Как перечислить все классификаторы scikit-learn, которые поддерживают pred_proba()

Мне нужен список всех классификаторов scikit-learn, которые поддерживают predict_proba(). Поскольку в документации нет простого способа получить эту информацию, как получить ее программно?

ОТВЕТЫ

Ответ 1

from sklearn.utils.testing import all_estimators

estimators = all_estimators()

for name, class_ in estimators:
    if hasattr(class_, 'predict_proba'):
        print(name)

Вы также можете использовать CalibratedClassifierCV, чтобы сделать любой классификатор одним, у которого есть predict_proba.

Это было задано ранее на SO, но я не могу его найти, поэтому вам следует извинить за дубликат;)

Ответ 2

AdaBoostClassifier

BaggingClassifier

BayesianGaussianMixture

BernoulliNB

CalibratedClassifierCV

ComplementNB

DecisionTreeClassifier

ExtraTreeClassifier

ExtraTreesClassifier

GaussianMixture

GaussianNB

GaussianProcessClassifier

GradientBoostingClassifier

KNeighborsClassifier

LabelPropagation

LabelSpreading

LinearDiscriminantAnalysis

Логистическая регрессия

LogisticRegressionCV

MLPClassifier

MultinomialNB

NuSVC

QuadraticDiscriminantAnalysis

RandomForestClassifier

SGDClassifier

SVC

_BinaryGaussianProcessClassifierLaplace

_ConstantPredictor