如何列出支持predict_proba()的所有scikit-learn分类器

Question

我需要一个支持predict_proba()方法的所有scikit-learn分类器的列表.由于文档没有提供获取该信息的简单方法,如何以编程方式获取此信息？

Answer 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之前被问到的,但我找不到它,所以你应该原谅副本;)