情节KMeans模型

编程问答 2022-08-11

问题描述

我想绘制KMeans ML的质心，但我不明白。添加图形图片。

from sklearn.cluster import KMeans
from sklearn import metrics
from sklearn.model_selection import train_test_split
y = np.radians(df["LATITUD"]) #series
x =  np.radians(df[["LONGITUD"]]) #dataframe
x_train,x_test,y_train,y_test = train_test_split(x,y,test_size=0.30,random_state=0)
KMeans =cluster.KMeans(n_clusters=5)
model2 = KMeans.fit(x_train,y_train)
model2 = KMeans.fit_predict(x_test)
centroids2 = np.array(KMeans.cluster_centers_)
labels = KMeans.labels_ 
K = 5
figure = plt.figure(figsize=(10,7))
plt.scatter(x_train,cmap="plasma",linewidths=1);
for k in range(K):
  plt.scatter(centroids2[k,0],centroids2[k,1],s=100,marker='*',color='red')

plt.show()

输出： IndexError Traceback（最近一次通话） 在（）中 3 plt.scatter（x_train，y_train，cmap =“ plasma”，线宽= 1）; 对于范围（k）中的k，为4： ----> 5个散点图（centroids2 [k，0]，centroids2 [k，1]，s = 100，marker ='*'，color ='red'） 6个show（） IndexError：索引1超出了尺寸1的轴1的范围

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

k-meansmatplotlibpython