1. Learn
    2. /
  2. Cursuri
    3. /
  3. Segmentarea clienților în Python
Connected

exercițiu

Calculează suma pătratelor erorilor

În acest exercițiu, vei calcula suma pătratelor erorilor pentru diferite numere de clustere, de la 1 la 15. În acest exemplu, folosim un set de date creat special pentru a obține o curbă „elbow" mai clară.

Am încărcat versiunea normalizată a datelor ca data_normalized. Modulul KMeans din scikit-learn este deja importat. De asemenea, am inițializat un dicționar gol pentru a stoca suma pătratelor erorilor: sse = {}.

Explorează datele în consolă dacă dorești.

Instrucțiuni

100 XP
  • Antrenează KMeans și calculează SSE pentru fiecare k cu valori între 1 și 15.
  • Inițializează KMeans cu k clustere și starea aleatoare 1.
  • Antrenează KMeans pe setul de date normalizat.
  • Atribuie suma pătratelor distanțelor elementului k din dicționarul sse.