from sklearn.cluster import KMeans
import numpy as np
# 生成一些示例数据
X = np.array([[1, 2], [1, 4], [1, 0],
[10, 2], [10, 4], [10, 0]])
# 创建KMeans模型,指定聚类数量为2
kmeans = KMeans(n_clusters=2, random_state=0).fit(X)
# 输出聚类中心
print("聚类中心:", kmeans.cluster_centers_)
# 预测新数据点的所属类别
new_data = np.array([[0, 0], [12, 3]])
predictions = kmeans.predict(new_data)
print("新数据点预测类别:", predictions)
# 输出每个样本的聚类标签
print("样本聚类标签:", kmeans.labels_)
sklearn.cluster 中导入 KMeans 类,并导入 numpy 用于处理数组。X,表示要进行聚类的数据点。KMeans 类创建一个聚类模型,并指定聚类的数量为2。random_state=0 确保每次运行结果一致。fit(X) 方法对数据进行聚类。cluster_centers_ 属性获取每个聚类的中心点。predict 方法对新的数据点进行分类预测。labels_ 属性获取每个样本所属的聚类标签。希望这段代码和解释对你有帮助!
上一篇:python venv 指定版本
下一篇:python数组的定义
Laravel PHP 深圳智简公司。版权所有©2023-2043 LaravelPHP 粤ICP备2021048745号-3
Laravel 中文站