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

iterate over k values and plot the inertia values for each k

josh.ipynb answered on June 9, 2022 Popularity 1/10 Helpfulness 1/10

answer iterate over k values and plot the inertia values for each k

iterate over k values and plot the inertia values for each k

Comment

ks = range(1, 6)
inertias = []

for k in ks:
    # Create a KMeans instance with k clusters: model
    model = KMeans(n_clusters=k)
    
    # Fit model to samples
    model.fit(samples)
    
    # Append the inertia to the list of inertias
    inertias.append(model.inertia_)
    
# Plot ks vs inertias
plt.plot(ks, inertias, '-o')
plt.xlabel('number of clusters, k')
plt.ylabel('inertia')
plt.xticks(ks)
plt.show()

Popularity 1/10 Helpfulness 1/10 Language python

Source: campus.datacamp.com

Tags: each iterate k plot python

Link to this answer
Share Copy Link