2017-02-10 8 views
1

По аналогии с How to list all scikit-learn classifiers that support predict_proba() Я хочу получить список всех алгоритмов классификации/регрессии/кластеризации, которые в настоящее время поддерживаются в scikit-learn.Как перечислить все алгоритмы классификации/регрессии/кластеризации в scikit-learn?

+0

Возможный дубликат https://stackoverflow.com/questions/41844311/list-of-all-classification-algorithms/41853264#41853264 –

+0

Вы прочитали мой вопрос? Это именно то, о чем я говорю и указываю на разницу. – DreamFlasher

ответ

2

Объединение How to list all scikit-learn classifiers that support predict_proba() и 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.

 Смежные вопросы

  • Нет связанных вопросов^_^