CommencezCommencez gratuitement

Visualiser la complexité du modèle

Vous avez maintenant calculé la justesse du modèle KNN sur les ensembles d'entraînement et de test en utilisant différentes valeurs de n_neighbors. Vous pouvez donc créer une courbe de complexité du modèle pour visualiser comment la performance évolue à mesure que le modèle devient moins complexe!

Les variables neighbors, train_accuracies et test_accuracies, que vous avez générées dans l'exercice précédent, ont été préchargées pour vous. Vous allez tracer les résultats afin de trouver le nombre optimal de voisins pour votre modèle.

Cette activité fait partie du cours

Apprentissage supervisé avec scikit-learn

Voir le cours

Instructions de l’exercice

  • Ajoutez le titre "KNN: Varying Number of Neighbors".
  • Tracez la méthode .values() de train_accuracies sur l'axe des y contre neighbors sur l'axe des x, avec l'étiquette "Training Accuracy".
  • Tracez la méthode .values() de test_accuracies sur l'axe des y contre neighbors sur l'axe des x, avec l'étiquette "Testing Accuracy".
  • Affichez le graphique.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Add a title
plt.title("____")

# Plot training accuracies
plt.plot(____, ____, label="____")

# Plot test accuracies
plt.plot(____, ____, label="____")

plt.legend()
plt.xlabel("Number of Neighbors")
plt.ylabel("Accuracy")

# Display the plot
____
Modifier et exécuter le code