python学习笔记（30）----sklearn-k-means

今天终于不是自闭的一天了，只调参还是舒服的呀，用

from sklearn.cluster import KMeans
import numpy as np
import matplotlib.pyplot as plt

data=np.genfromtxt('kmeans.txt',delimiter=' ')
k=4

model=KMeans(n_clusters=k)
model.fit(data)
#分类质点
centers=model.cluster_centers_
print(centers)
#预测结果
result=model.predict(data)


#画图
mark=['or','ob','oy','og']
for i,j in enumerate(data):
    plt.plot(j[0],j[1],mark[result[i]])

mark=['*r','*b','*y','*g']
for i,j in enumerate(centers):
    plt.plot(j[0],j[1],mark[i],markersize=10)


plt.show()