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;)