from sklearn.cluster import KMeans kme = KMeans(n_clusters=10) model = kme.fit(x_train,y_train) y_pre = kme.predict(x_test)
模型评估
ARI指标
1 2
from sklearn.metrics import adjusted_rand_score adjusted_rand_score(y_test,y_pre)
0.66305779493265249
轮廓系数
1 2
from sklearn.metrics import silhouette_score silhouette_score(y_test.values.reshape(-1,1),y_pre.reshape(-1,1),metric="euclidean")
c:\users\qiank\appdata\local\programs\python\python35\lib\site-packages\sklearn\utils\validation.py:547: DataConversionWarning: A column-vector y was passed when a 1d array was expected. Please change the shape of y to (n_samples, ), for example using ravel().
y = column_or_1d(y, warn=True)
0.27296875226980805