如何在scikit-learn中列出所有分类/回归/聚类算法？

Question

类似于如何列出所有支持predict_proba（）的scikit学习分类器，我想检索scikit-learn当前支持的所有分类/回归/聚类算法的列表。

Answer 1

结合如何列出所有支持predict_proba（）和http://scikit-learn.org/stable/modules/classes.html#module-sklearn.base的scikit-learn分类器可得出解决方案：

from sklearn.utils.testing import all_estimators
from sklearn import base

estimators = all_estimators()

for name, class_ in estimators:
    if issubclass(class_, base.ClassifierMixin):
        print(name)

或使用任何其他基类：ClusterMixin，RegressorMixin，TransformerMixin。

Answer 2

作为更新的解决方案，sklearn 将模块更新为sklearn.utils.all_estimators。以下是导入所有回归模型的示例：

from sklearn.utils import all_estimators

estimators = all_estimators(type_filter='regressor')

all_regs = []
for name, RegressorClass in estimators:
    try:
        print('Appending', name)
        reg = RegressorClass()
        all_regs.append(reg)
    except Exception as e:
        print(e)

其中一些需要初始化参数（如估计器），并且必须使用 try..except 来忽略。