Hello all,
I've written some code to run Kmeans on a data set (please see below).
And I've plotted the results, with my two clusters/ centroids.
However, I've to re-run Kmeans several times and pull up different plots
(showing the different centroid positions).
Can someone point me in the right direction how to write this extra code to
perform this task?
Then I've to conclude if Kmeans is stable. I believe this is the lowest sum of
squared errors?
Thanking you in advance.
#pandas used to read dataset and return the data
#numpy and matplotlib to represent and visualize the data
#sklearn to implement kmeans algorithm
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans
from sklearn.preprocessing import StandardScaler
#import the data
data = pd.read_csv('file.csv')
#extract values
x=data['V1']
y=data['V2']
V1_V2 = np.column_stack ((V1, V2))
km_res = KMeans (n_clusters= 2).fit(V1_V2)
y_kmeans = km_res.predict(V1_V2)
plt.scatter(V1, V2, c=y_kmeans, cmap='viridis', s = 50, alpha = 0.5)
plt.xlabel('V1')
plt.ylabel('V2')
plt.title('Visualization of raw data');
clusters = km_res.cluster_centers_
plt.scatter(clusters[:,0], clusters[:,1], c='blue', s=150)
Get Outlook for iOS<https://aka.ms/o0ukef>
_______________________________________________
scikit-learn mailing list
scikit-learn@python.org
https://mail.python.org/mailman/listinfo/scikit-learn
