Lorsque j'utilise randomforest dans sklearn pour faire une classification binaire. Je sais que je peux utiliser clf.predict(X) pour obtenir la classe prédite. Et quand j'utilise clf.predict_proba(X) j'obtiens un tableau qui ressemble à ceci : 
Je pense que la première colonne indique la probabilité de la prédiction ? Comment puis-je obtenir une colonne qui correspond à la probabilité que la classe soit 1 ?
De la RandomForest.predict_proba docs :
predict_proba(X)Prédire les probabilités de classe pour X.
Renvoie à :
p: [...] Les probabilités de classe des échantillons d'entrée. L'ordre des classes de l'ordre des classes correspond à celui de l'attributclasses_.
Vous pouvez consulter le clf.classes_ pour voir à quel indice votre classe 1 apparaît, et il suffit ensuite d'accéder aux probabilités ainsi :
prob_class_1 = clf.predict_proba(X)[:, i] Où i est l'indice de la classe 1 en clf.classes_ .
Prograide est une communauté de développeurs qui cherche à élargir la connaissance de la programmation au-delà de l'anglais.
Pour cela nous avons les plus grands doutes résolus en français et vous pouvez aussi poser vos propres questions ou résoudre celles des autres.
