CommencezCommencez gratuitement

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

Voir le cours

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_predictions pour 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(____[____]))
Modifier et exécuter le code