聚类算法的python

  本程序由本作者方便自我记忆

import numpy as np
import matplotlib.pyplot as plt
from sklearn import datasets
iris = datasets.load_iris()
from sklearn.model_selection import train_test_split
x_train, x_test,y_train,y_test= train_test_split(iris.data, iris.target)     # 导入鸢尾花数据集


from sklearn.cluster import KMeans
model = KMeans(n_clusters=5)      # kMeans包
model.fit(x_train)
y_pre=model.predict(x_test)

# KMeans.fit(x_train)
from sklearn import metrics
print(metrics.adjusted_rand_score(y_test,y_pre))

#from sklearn.metrics import silhouestte_score
plt.subplot(3,2,1)