Kom igångKom igång gratis

Hur många kluster av spannmål?

I videon lärde du dig hur man väljer ett lämpligt antal kluster för en datamängd med hjälp av k-means tröghetsgraf. Du får en array samples som innehåller mätningar (till exempel area, omkrets, längd och flera andra) av spannmålsprover. Vilket antal kluster är lämpligt i det här fallet?

KMeans och PyPlot (plt) har redan importerats åt dig.

Den här datamängden är hämtad från UCI Machine Learning Repository.

Den här övningen är en del av kursen

Oövervakad inlärning i Python

Visa kurs

Övningsinstruktioner

  • Utför följande steg för vart och ett av de angivna värdena på k:
  • Skapa en KMeans-instans kallad model med k kluster.
  • Anpassa modellen till spannmålsdata i samples.
  • Lägg till värdet på attributet inertia_ från model i listan inertias.
  • Koden för att plotta ks mot inertias är redan skriven, så klicka på Skicka in svar för att se grafen!

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

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

for k in ks:
    # Create a KMeans instance with k clusters: model
    ____
    
    # Fit model to samples
    ____
    
    # Append the inertia to the list of inertias
    ____
    
# Plot ks vs inertias
plt.plot(ks, inertias, '-o')
plt.xlabel('number of clusters, k')
plt.ylabel('inertia')
plt.xticks(ks)
plt.show()
Redigera och kör kod