类似于如何列出所有支持predict_proba()的scikit-learn分类器,我想获取scikit-learn当前支持的所有分类/回归/聚类算法的列表。
回答:
结合如何列出所有支持predict_proba()的scikit-learn分类器和http://scikit-learn.org/stable/modules/classes.html#module-sklearn.base可以得到解决方案:
from sklearn.utils.testing import all_estimatorsfrom sklearn import baseestimators = all_estimators()for name, class_ in estimators: if issubclass(class_, base.ClassifierMixin): print(name)
或者使用其他基础类:ClusterMixin, RegressorMixin, TransformerMixin。