Prédictions de classification
En validation de modèles, il est souvent important d'en savoir plus sur les prédictions que la simple classe finale. Quand on prédit l'équipe gagnante d'un match, on veut généralement aussi connaître à quel point cette victoire est probable.
| Probability | Prediction | Meaning |
|---|---|---|
| 0 < .50 | 0 | Team Loses |
| .50 + | 1 | Team Wins |
Dans cet exercice, vous allez utiliser les méthodes .predict() et .predict_proba() avec l'ensemble de données tic_tac_toe. La première méthode donne une prédiction indiquant si le Joueur 1 gagnera la partie, et la seconde fournit la probabilité que le Joueur 1 gagne. Utilisez rfc comme modèle de classification par forêts aléatoires.
Cette activité fait partie du cours
Validation de modèles en Python
Instructions de l’exercice
- Créez deux tableaux de prédictions : un pour les valeurs de classification et un pour les probabilités prédites.
- Utilisez la méthode
.value_counts()d'une Series pandas pour afficher le nombre d'observations attribuées à chaque classe. - Affichez la première observation de
probability_predictionspour voir comment les probabilités sont structurées.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Fit the rfc model.
rfc.fit(X_train, y_train)
# Create arrays of predictions
classification_predictions = rfc.____(X_test)
probability_predictions = rfc.____(X_test)
# Print out count of binary predictions
print(pd.Series(____).____())
# Print the first value from probability_predictions
print('The first predicted probabilities are: {}'.format(____[____]))