2015-05-05 6 views

risposta

17
from sklearn.utils.testing import all_estimators 

estimators = all_estimators() 

for name, class_ in estimators: 
    if hasattr(class_, 'predict_proba'): 
     print(name) 

È inoltre possibile utilizzare CalibratedClassifierCV per fare qualsiasi classificatore in uno che ha predict_proba.

Questo è stato chiesto prima su SO, ma non riesco a trovarlo, così si dovrebbe essere scusato per il duplicato;)

+0

Grazie! Ho letto solo ieri su CalibratedClassifierCV e sono ansioso di provarlo. – Toby

+0

Oh, e mesi dopo, mi sono imbattuto in quello che potrebbe essere il duplicato che avevi in ​​mente: http://stackoverflow.com/questions/22737053/which-estimators-in-scikit-learn-dont-support-sparse-matrices – Toby

+0

Questo non è disponibile nella versione 0.19. La documentazione si riferisce ancora ad essa http://scikit-learn.org/stable/developers/utilities.html#testing-functions ma la funzione di test non funziona. C'è un altro modo per sapere se un metodo è supportato su tutti gli stimatori (come domanda originale)? – iblasi

Problemi correlati