使用K-means对sklearn自带数据集进行聚类,使用sklearn,并且对K值进行寻找最优 全部代码

时间: 2024-03-05 11:48:51 浏览: 17
好的,下面是使用K-means对sklearn自带数据集进行聚类的完整代码,包括了K值的寻找最优。 ```python import numpy as np import matplotlib.pyplot as plt from sklearn.cluster import KMeans from sklearn.datasets import make_blobs from sklearn.metrics import silhouette_score # 生成数据 X, y = make_blobs(n_samples=1000, n_features=2, centers=4, random_state=42) # 寻找最优K值 sse = [] silhouette_coef = [] for k in range(2, 11): kmeans = KMeans(n_clusters=k, random_state=0).fit(X) sse.append(kmeans.inertia_) score = silhouette_score(X, kmeans.labels_, metric='euclidean') silhouette_coef.append(score) # 绘制SSE和轮廓系数曲线 plt.figure(figsize=(12, 6)) plt.subplot(121) plt.plot(range(2, 11), sse, marker='o') plt.xlabel('Number of clusters') plt.ylabel('SSE') plt.title('Elbow Method') plt.subplot(122) plt.plot(range(2, 11), silhouette_coef, marker='o') plt.xlabel('Number of clusters') plt.ylabel('Silhouette Coefficient') plt.title('Silhouette Method') plt.show() # 选取最优K值进行聚类 k = 4 kmeans = KMeans(n_clusters=k, random_state=0).fit(X) # 可视化聚类结果 centers = kmeans.cluster_centers_ colors = ['r', 'g', 'b', 'y', 'c', 'm'] for i in range(k): index = np.where(kmeans.labels_ == i) plt.scatter(X[index, 0], X[index, 1], c=colors[i], label='Cluster {}'.format(i)) plt.scatter(centers[:, 0], centers[:, 1], marker='*', s=200, c='#000000') plt.legend() plt.xlabel('Feature 1') plt.ylabel('Feature 2') plt.show() ``` 在代码中,首先使用sklearn中的make_blobs函数生成了一个聚类数据集,然后使用循环遍历K值,通过KMeans算法进行聚类,并计算SSE和轮廓系数。最后,根据SSE和轮廓系数的曲线,选取最优K值进行聚类,并可视化聚类结果。 需要注意的是,由于该数据集是人为生成的,因此不同的数据集可能需要不同的K值才能得到最优聚类结果。因此,在实际应用中还需要根据具体数据集进行调整。

相关推荐

最新推荐

recommend-type

详解Java实现的k-means聚类算法

主要介绍了详解Java实现的k-means聚类算法,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
recommend-type

python 代码实现k-means聚类分析的思路(不使用现成聚类库)

主要介绍了python 代码实现k-means聚类分析(不使用现成聚类库),本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
recommend-type

python基于K-means聚类算法的图像分割

主要介绍了python基于K-means聚类算法的图像分割,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
recommend-type

Python——K-means聚类分析及其结果可视化

K-Means是聚类算法的一种,通过距离来判断数据点间的相似度并据此对数据进行聚类。 1 聚类算法 科学计算中的聚类方法 方法名称 参数 可伸缩性 用例 几何形状(使用的指标) K-Means number of ...
recommend-type

k-means 聚类算法与Python实现代码

k-means 聚类算法思想先随机选择k个聚类中心,把集合里的元素与最近的聚类中心聚为一类,得到一次聚类,再把每一个类的均值作为新的聚类中心重新聚类,迭代n次得到最终结果分步解析 一、初始化聚类中心 首先随机...
recommend-type

zigbee-cluster-library-specification

最新的zigbee-cluster-library-specification说明文档。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

实现实时数据湖架构:Kafka与Hive集成

![实现实时数据湖架构:Kafka与Hive集成](https://img-blog.csdnimg.cn/img_convert/10eb2e6972b3b6086286fc64c0b3ee41.jpeg) # 1. 实时数据湖架构概述** 实时数据湖是一种现代数据管理架构,它允许企业以低延迟的方式收集、存储和处理大量数据。与传统数据仓库不同,实时数据湖不依赖于预先定义的模式,而是采用灵活的架构,可以处理各种数据类型和格式。这种架构为企业提供了以下优势: - **实时洞察:**实时数据湖允许企业访问最新的数据,从而做出更明智的决策。 - **数据民主化:**实时数据湖使各种利益相关者都可
recommend-type

SPDK_NVMF_DISCOVERY_NQN是什么 有什么作用

SPDK_NVMF_DISCOVERY_NQN 是 SPDK (Storage Performance Development Kit) 中用于查询 NVMf (Non-Volatile Memory express over Fabrics) 存储设备名称的协议。NVMf 是一种基于网络的存储协议,可用于连接远程非易失性内存存储器。 SPDK_NVMF_DISCOVERY_NQN 的作用是让存储应用程序能够通过 SPDK 查询 NVMf 存储设备的名称,以便能够访问这些存储设备。通过查询 NVMf 存储设备名称,存储应用程序可以获取必要的信息,例如存储设备的IP地址、端口号、名称等,以便能
recommend-type

JSBSim Reference Manual

JSBSim参考手册,其中包含JSBSim简介,JSBSim配置文件xml的编写语法,编程手册以及一些应用实例等。其中有部分内容还没有写完,估计有生之年很难看到完整版了,但是内容还是很有参考价值的。