如何在scikit-learn中列出所有分类/回归/聚类算法? [英] How to list all classification/regression/clustering algorithms in scikit-learn?
本文介绍了如何在scikit-learn中列出所有分类/回归/聚类算法?的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
类似于如何列出全部支持predict_proba()的scikit-learn分类器我想检索scikit-learn当前支持的所有分类/回归/聚类算法的列表.
In analogy to How to list all scikit-learn classifiers that support predict_proba() I want to retrieve a list of all classification/regression/clustering algorithms currently supported in scikit-learn.
推荐答案
结合 http://scikit-learn.org/stable/modules/classes.html#module-sklearn.base 产生解决方案:
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.
Or use any other base class: ClusterMixin, RegressorMixin, TransformerMixin.
这篇关于如何在scikit-learn中列出所有分类/回归/聚类算法?的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
查看全文