Tracer la somme des erreurs au carré
Vous allez maintenant tracer la somme des erreurs au carré pour chaque valeur de k et vérifier s'il y a un coude. Cela vous aidera à déterminer le nombre de groupes recommandé.
La somme des erreurs au carré est chargée sous forme de dictionnaire nommé sse depuis l'exercice précédent. matplotlib.pyplot a été importé sous le nom plt, et seaborn sous sns.
Vous pouvez explorer le dictionnaire dans la console.
Cette activité fait partie du cours
Segmentation de la clientèle en Python
Instructions de l’exercice
- Ajoutez le titre du graphique « The Elbow Method ».
- Ajoutez l'étiquette de l'axe des X « k ».
- Ajoutez l'étiquette de l'axe des Y « SSE ».
- Tracez les valeurs de SSE pour chaque
kstocké comme clé dans le dictionnaire.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Add the plot title "The Elbow Method"
plt.____('The Elbow Method')
# Add X-axis label "k"
plt.____('____')
# Add Y-axis label "SSE"
plt.____('____')
# Plot SSE values for each key in the dictionary
sns.____(x=list(sse.____()), y=list(sse.____()))
plt.show()