python - machine - Cómo enumerar todos los clasificadores de scikit-learn compatibles con predic_proba()

scikit-learn vs tensorflow

Question

Necesito una lista de todos los clasificadores scikit-learn que admiten el método predict_proba (). Dado que la documentación no proporciona una manera fácil de obtener esa información, ¿cómo se puede obtener esto programáticamente?

Answer 1

from sklearn.utils.testing import all_estimators

estimators = all_estimators()

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

También puede usar CalibratedClassifierCV para convertir cualquier clasificador en uno que tenga predict_proba .

Esto se solicitó anteriormente en SO, pero no puedo encontrarlo, por lo que debes ser excusado por el duplicado;)