python sklearn k-means 代码示例

好的，以下是用 sklearn 库中的 KMeans 模型实现 K 均值聚类的代码示例：

from sklearn.cluster import KMeans
import numpy as np

# 创建数据集
X = np.array([[1, 2], [1, 4], [1, 0], [4, 2], [4, 4], [4, 0]])

# 创建 KMeans 模型，并设置聚类数量为 2
kmeans = KMeans(n_clusters=2)

# 训练模型
kmeans.fit(X)

# 预测每个样本所属的聚类
predictions = kmeans.predict(X