Breaking News: Grepper is joining You.com. Read the official announcement!

elbow method k means sklearn

Juan David Morales Navarrete answered on March 25, 2022 Popularity 8/10 Helpfulness 10/10

answer elbow method k means sklearn

elbow method k means sklearn

Comment

from sklearn.cluster import KMeans
from matplotlib import pyplot as plt

X = # <your_data>
distorsions = []
for k in range(2, 20):
    kmeans = KMeans(n_clusters=k)
    kmeans.fit(X)
    distorsions.append(kmeans.inertia_)

fig = plt.figure(figsize=(15, 5))
plt.plot(range(2, 20), distorsions)
plt.grid(True)
plt.title('Elbow curve')

Popularity 8/10 Helpfulness 10/10 Language python

Source: stackoverflow.com

Tags: k-means python

Link to this answer
Share Copy Link