[問題] cluster analysis畫圖

作者: iSad56 (窩南果)   2018-04-18 17:49:35
各位大大安安
小弟在摸索用kmeans聚類 在網路上參考一些範例
import numpy as np
from sklearn.cluster import KMeans
list1 = [88.0,74.0,96.0,85.0]
list2 = [92.0,99.0,95.0,94.0]
list3 = [91.0,87.0,99.0,95.0]
list4 = [78.0,99.0,97.0,81.0]
list5 = [88.0,78.0,98.0,84.0]
list6 = [100.0,95.0,100.0,92.0]
X = np.array([list1,list2,list3,list4,list5,list6])
kmeans = KMeans(n_clusters = 2).fit(X)
pred = kmeans.predict(X)
print(pred)
Output:
[0 1 1 1 0 1]
是成功分出兩群了
但是爬文的過程 看到一些很漂亮的圖 像這樣
https://upload.cc/i1/2018/04/18/pFLHgX.png
完全看不懂這個X軸跟Y軸是怎麼放的
如果以上面的範例來說 應該要怎麼畫出圖呢?
感謝
作者: gloneer (綠蠵龜)   2018-04-18 18:38:00
看著很像用pyplot的scatter畫出來的,不知道感覺有沒有錯( ^ω^)
作者: Mchord (Mchord)   2018-04-18 19:05:00
你的資料是6維的,不好呈現,如果利用PCA之類的先降到2維就可以畫眼花了…你是6筆4維的這篇可參考看看 https://tinyurl.com/y984mnq7
作者: bestchiao (呼拉拉)   2018-04-19 09:40:00
用scatter plot然後顏色用你預測出來的label文章中的類別 指的就是資料實際的label 共有9個 所以輸出的顏色有9種但注意文章中只有用pca 沒有用到kmeans
作者: Mchord (Mchord)   2018-04-19 20:36:00
kmeans分群完後,再把每筆資料歸屬group的index視做該資料的類別

Links booklink

Contact Us: admin [ a t ] ucptt.com