机器学习之 K-means算法的代码实现

2024-08-28 00:52

本文主要是介绍机器学习之 K-means算法的代码实现,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

K-means 算法简介

K-means 是一种常用的无监督学习算法,主要用于数据聚类。它的主要思想是将数据集中的数据分成 K 个簇(Cluster),使得簇内的数据点尽可能相似,而簇间的差异尽可能大。K-means 算法的核心步骤包括初始化质心、分配数据点到最近的质心、更新质心,重复这个过程直到收敛。

导入实验数据

在本实验中,我们将使用一个啤酒数据集,数据集包含每种啤酒的卡路里(calories)、钠含量(sodium)、酒精含量(alcohol)和价格(cost)。这些数据将用于 K-means 聚类分析。

实验数据如下:

实验代码详解

导入库和数据

import pandas as pd
from sklearn.cluster import KMeans
from sklearn import metrics# 读取文件
beer = pd.read_table("data.txt", sep=' ', encoding='utf8', engine='python')# 传入变量(列名)
X = beer[["calories", "sodium", "alcohol", "cost"]]
print(X)

解释:

  • 我们首先导入所需的库。
  • 使用 pd.read_table 函数读取名为 data.txt 的数据文件,其中 sep=' ' 表示数据分隔符为空格。
  • 然后选择四列数据作为输入特征,并将其存储在变量 X 中。

计算轮廓系数得分 

scores = []
for k in range(2, 10):  # 寻找合适的K值labels = KMeans(n_clusters=k).fit(X).labels_score = metrics.silhouette_score(X, labels)  # 轮廓系数scores.append(score)
print(scores)

解释:

  • 这部分代码的目的是找到最适合的簇数量。
  • 使用 range(2, 10) 来循环尝试从 2 到 9 个簇。
  • 对于每个簇数量,我们使用 KMeans 类进行聚类,并获取聚类标签。
  • 使用 metrics.silhouette_score 计算轮廓系数得分,该得分越高表示簇内的数据点越相似,簇间差异越大。
  • 得分被存储在列表 scores 中。

绘制得分结果

import matplotlib.pyplot as pltplt.plot(list(range(2, 10)), scores)
plt.xlabel("Number of Clusters Initialized")
plt.ylabel("Silhouette Score")
plt.show()

解释:

  • 使用 matplotlib 绘制簇数量与轮廓系数得分之间的关系。
  • 通过观察得分曲线,我们可以找到最佳的簇数量。

聚类

km = KMeans(n_clusters=2).fit(X)  # K值为2【分为2类】
beer['cluster'] = km.labels_

解释:

  • 根据之前的得分结果,我们选择了 K=2 进行聚类。
  • 使用 KMeans 类进行聚类,并将结果存储在 beer DataFrame 的新列 cluster 中。

对聚类结果进行评分

score = metrics.silhouette_score(X, beer.cluster)
print(score)

解释:

  • 最后,我们再次计算轮廓系数得分,以评估最终聚类结果的质量。
输出结果:

总结

通过上述步骤,我们完成了 K-means 聚类分析的过程。K-means 算法通过迭代的方式不断优化簇内数据的相似性和簇间数据的差异性,从而达到聚类的目的。在本实验中,我们不仅实现了 K-means 算法,还通过轮廓系数得分来评估不同簇数量下的聚类效果,并最终确定了最佳的簇数量。这种方法在很多实际应用中都非常有用,特别是在需要对数据进行分组和分析的场景中。

这篇关于机器学习之 K-means算法的代码实现的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1113264

相关文章

HarmonyOS学习(七)——UI(五)常用布局总结

自适应布局 1.1、线性布局(LinearLayout) 通过线性容器Row和Column实现线性布局。Column容器内的子组件按照垂直方向排列,Row组件中的子组件按照水平方向排列。 属性说明space通过space参数设置主轴上子组件的间距,达到各子组件在排列上的等间距效果alignItems设置子组件在交叉轴上的对齐方式,且在各类尺寸屏幕上表现一致,其中交叉轴为垂直时,取值为Vert

Ilya-AI分享的他在OpenAI学习到的15个提示工程技巧

Ilya(不是本人,claude AI)在社交媒体上分享了他在OpenAI学习到的15个Prompt撰写技巧。 以下是详细的内容: 提示精确化:在编写提示时,力求表达清晰准确。清楚地阐述任务需求和概念定义至关重要。例:不用"分析文本",而用"判断这段话的情感倾向:积极、消极还是中性"。 快速迭代:善于快速连续调整提示。熟练的提示工程师能够灵活地进行多轮优化。例:从"总结文章"到"用

不懂推荐算法也能设计推荐系统

本文以商业化应用推荐为例,告诉我们不懂推荐算法的产品,也能从产品侧出发, 设计出一款不错的推荐系统。 相信很多新手产品,看到算法二字,多是懵圈的。 什么排序算法、最短路径等都是相对传统的算法(注:传统是指科班出身的产品都会接触过)。但对于推荐算法,多数产品对着网上搜到的资源,都会无从下手。特别当某些推荐算法 和 “AI”扯上关系后,更是加大了理解的难度。 但,不了解推荐算法,就无法做推荐系

【前端学习】AntV G6-08 深入图形与图形分组、自定义节点、节点动画(下)

【课程链接】 AntV G6:深入图形与图形分组、自定义节点、节点动画(下)_哔哩哔哩_bilibili 本章十吾老师讲解了一个复杂的自定义节点中,应该怎样去计算和绘制图形,如何给一个图形制作不间断的动画,以及在鼠标事件之后产生动画。(有点难,需要好好理解) <!DOCTYPE html><html><head><meta charset="UTF-8"><title>06

学习hash总结

2014/1/29/   最近刚开始学hash,名字很陌生,但是hash的思想却很熟悉,以前早就做过此类的题,但是不知道这就是hash思想而已,说白了hash就是一个映射,往往灵活利用数组的下标来实现算法,hash的作用:1、判重;2、统计次数;

hdu1043(八数码问题,广搜 + hash(实现状态压缩) )

利用康拓展开将一个排列映射成一个自然数,然后就变成了普通的广搜题。 #include<iostream>#include<algorithm>#include<string>#include<stack>#include<queue>#include<map>#include<stdio.h>#include<stdlib.h>#include<ctype.h>#inclu

康拓展开(hash算法中会用到)

康拓展开是一个全排列到一个自然数的双射(也就是某个全排列与某个自然数一一对应) 公式: X=a[n]*(n-1)!+a[n-1]*(n-2)!+...+a[i]*(i-1)!+...+a[1]*0! 其中,a[i]为整数,并且0<=a[i]<i,1<=i<=n。(a[i]在不同应用中的含义不同); 典型应用: 计算当前排列在所有由小到大全排列中的顺序,也就是说求当前排列是第

csu 1446 Problem J Modified LCS (扩展欧几里得算法的简单应用)

这是一道扩展欧几里得算法的简单应用题,这题是在湖南多校训练赛中队友ac的一道题,在比赛之后请教了队友,然后自己把它a掉 这也是自己独自做扩展欧几里得算法的题目 题意:把题意转变下就变成了:求d1*x - d2*y = f2 - f1的解,很明显用exgcd来解 下面介绍一下exgcd的一些知识点:求ax + by = c的解 一、首先求ax + by = gcd(a,b)的解 这个

综合安防管理平台LntonAIServer视频监控汇聚抖动检测算法优势

LntonAIServer视频质量诊断功能中的抖动检测是一个专门针对视频稳定性进行分析的功能。抖动通常是指视频帧之间的不必要运动,这种运动可能是由于摄像机的移动、传输中的错误或编解码问题导致的。抖动检测对于确保视频内容的平滑性和观看体验至关重要。 优势 1. 提高图像质量 - 清晰度提升:减少抖动,提高图像的清晰度和细节表现力,使得监控画面更加真实可信。 - 细节增强:在低光条件下,抖

【数据结构】——原来排序算法搞懂这些就行,轻松拿捏

前言:快速排序的实现最重要的是找基准值,下面让我们来了解如何实现找基准值 基准值的注释:在快排的过程中,每一次我们要取一个元素作为枢纽值,以这个数字来将序列划分为两部分。 在此我们采用三数取中法,也就是取左端、中间、右端三个数,然后进行排序,将中间数作为枢纽值。 快速排序实现主框架: //快速排序 void QuickSort(int* arr, int left, int rig