Answers for "how to plot kmeans centroids"

0

how to plot kmeans centroids

#assuming we trained and saved the model in 'kmean_model' obj 
predict = kmean_model.predict(df)
plt.scatter(df.iloc[:,2],df.iloc[:,3],c = predict,cmap = 'viridis')
plt.scatter(kmean_model.cluster_centers_[:,2],kmean_model.cluster_centers_[:,3],c='red',marker = 'x')
Posted by: Guest on April-18-2022

Python Answers by Framework

Browse Popular Code Answers by Language