Kmeans

Question 1

train a model with n clusters, using df ‘x’

Answer 1

kmeans = KMeans(n)

kmeans.fit(x)

Question 2

see which cluster each data point in ‘x’ is assigned

Answer 2

identified_clusters = kmeans.fit_predict(x)
df['Cluster'] = identified_clusters

Question 3

what is the measure to use to determine number of k? what is it called? What’s the code?

Answer 3

kmeans.inertia_