Tracer la somme des erreurs au carré
Vous allez maintenant tracer la somme des erreurs au carré pour chaque valeur de k et voir s’il existe un coude. Cela vous aidera à déterminer le nombre de clusters 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.
Cet exercice fait partie du cours
Customer Segmentation in Python
Instructions
- Ajoutez le titre du graphique « The Elbow Method ».
- Ajoutez le libellé de l’axe des X « k ».
- Ajoutez le libellé 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 cet exemple de code.
# 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()