二乗誤差の合計をプロットする
ここでは、各 k の二乗誤差の合計をプロットし、エルボー(折れ曲がり)があるかを確認します。これにより、推奨されるクラスター数の目安がわかります。
二乗誤差の合計は、前の演習で sse という辞書として読み込まれています。matplotlib.pyplot は plt、seaborn は sns として読み込まれています。
コンソールでこの辞書を確認できます。
この演習はコースの一部です
Pythonで学ぶカスタマーセグメンテーション
演習の手順
- プロットのタイトルに "The Elbow Method" を追加します。
- X軸ラベルに "k" を追加します。
- Y軸ラベルに "SSE" を追加します。
- 辞書のキーとして保存されている各
kに対する SSE の値をプロットします。
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
# 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()