from sklearn.cluster import KMeans

distortions = []

# œKkT
for i in range(1, 11):
  km = KMeans(n_clusters=i, init="k-means++", n_init=10, max_iter=300, random_state=0)
  km.fit(pixels_pca)
  distortions.append(km.inertia_)

# eO[vSSE`
plt.plot(range(1, 11), distortions, marker="o")
plt.xticks(np.arange(1, 11, 1))
plt.xlabel("Number of groups")
plt.ylabel("SSE")
plt.show()
