参考:http://blog.topspeedsnail.com/archives/10349
Python版本:python3.6.2
Matlab版本:
一、K-Means算法的Python实现
这里以
泰坦尼克号遇难者名单为例,通过除survived以外字段进行聚类(k=2,生/死),然后再和survived进行对比。
#参考自:http://blog.topspeedsnail.com/archives/10349
import numpy as np
from sklearn.cluster import KMeans
from sklearn import prep