alexanderholt
11/14/2017 - 4:32 PM

K Means

k = 3
kmeans = cluster.KMeans(n_clusters=k)
kmeans.fit(X_scaled)

labels = kmeans.labels_
centroids = kmeans.cluster_centers_
inertia = kmeans.inertia_

print('Centroids:', centroids)
print('')
print('Inertia:', inertia)

print('Silhouette Score:', metrics.silhouette_score(X_scaled, labels, metric='euclidean'))

df['label'] = labels
X['label'] = labels
df

cols = X.columns[:-1]
sns.pairplot(X, x_vars=cols, y_vars= cols, hue='label')