机器学习EM原理和K-mean聚类

Posted 2023-04-05 无水先生

一、教程说明

        EM算法就是expect maxmise算法，就是“期望最大化”的缩写。本篇首先提出：1 什么是期望？ 2 期望最大化是个啥意思？k-mean聚类中如何用EM算法？ 

所涉及的概念：

1.   期望
2.   期望的加权平均理解
3.   概率模型和统计模型
4.   期望最大化
5.   k-mean算法的原理

二、什么是期望？

2.1 从一个思想实验入门

        在回答这个概念之前，我们可以做一个思想实验。

        假如：我们这里有一枚六面骰子

        1）每次掷出“1”奖励一块钱，那么掷出100次，您能得到几块钱？

        我们很容易想到：掷出100次，获得“1”的次数大约100/6次,每次的1块，总数大约1×100/6 =16.6元。故

        2）如果每次掷出“1”奖励一块钱，每掷出“2”奖励三元，那么掷出100次，您能得到几块钱？

        100次掷出，得到“2”的可能次数为100/6， 每次奖励3块，那么总的收益数为：

         3)更一般地，我们给出一个收益函数，那么，掷出100次后，总的收益是多少？

        给出收益函数：

        

        给出概率模型：

        

        那么，抛掷100次后，总的收益就是期望，它的公式为：

E(F)  =  <F, R >                     期望就是收益函数和概率模型的内积！

• 掷出100次的获益（期望）：

•  掷出1 次的获益（期望）

        结论：期望就是收益函数和概率模型的内积！此内积表示概率作用下的总收益。

 2.2 改变概率模型

        将上面的概率模型是：将其改成其它模型：

        期望可以如法炮制：

        结论： 期望就是对“多路贡献”的加权平均。加权平均就是“多路贡献”的总贡献。

2.3 给期望一个定义

• 所谓期望，就是一个收益函数与一个概率分布模型的内积。
• 所谓期望就是在一个收益函数上施加概率模型，最终获益的总量。
• 所谓期望，就是一个收益函数，在一个概率模型下的收益的概率平均（或加权平均）。

        注意：将收益函数赋予其它意义，期望还能有其它意义。

三、期望的几何意义

     我们看这里是教材对期望的定义：     

这个定义是否与上述的“收益函数”有矛盾？答曰“没有”，因为这里只要将“收益函数”定义为：

 F = X 就可以了。问题是这种期望有啥意义？

        期望的几何意义：一组数据的概率加权平均，或叫中心点，或叫重心。

四、期望的最大化原理（EM）

        引理：如果一组数据符合统计规律，那么这组数据能和理论上的一个概率模型对应。

 上图假定有两组数据，符合统计规律，且给出4个均匀分布概率模型。问两组数据分别和哪个理论模型匹配？

分别做内积：

组1和【1-4】期望：   【1，2，3，4】* 【1/4,1/4,1/4,1/4 】 = 2.5

组1和【2-5】期望：   【1，2，3，4，0】* 【0，1/4,1/4,1/4,1/4】 = 2.25

组1和【1-6】期望：   【1，2，3，4，0，0】* 【1/6，1/6,1/6,1/6,1/6,1/6】 =  1.666

因此：组1和【1-4】期望最大，说明组1和该分布最匹配。

组2和【12-17】的期望： 【12，13，14，15，16，17】* 【1/6，1/6,1/6,1/6,1/6,1/6】=14.5

结论：当一组数与一个概率模型匹配，那么它们的内积（期望）最大的。这就是期望最大原理。

五、K-mean算法原理

5.1 算法简要描述

kmeans的计算方法如下：

1  对于一组数据，我们假定需要将它们分成K 个类

2  随机选取k个中心点（期望）作为初始概率模型的位置（聚类重心）。

3  我们假设分类是合理的，遍历所有数据点，通过距离模型，将他们归到指定的概率模型中。

4  从以上的归类中，计算各分类的期望（平均值），修改聚类重心。并重新修改验证分类的合理性，并最近的中心点中。

5 重复3-4，直到这k个中心点不再变化（收敛了），或执行了足够多的迭代.

5.2 举个实际例子

给出如下数据集： Data =  2，3，4，10，11， 12，20，25，30

我们假定k=2（就是要分成两类）

随机给出两个点c1 = 9，c2=10，作为两个聚类中心（期望）

数据集（坐标）	到c1距离	到c2距离	距离比较	最后归类
2	7	8	<	c1
3	6	7	<	c1
4	5	6	<	c1
10	1	0	>	c2
11	2	1	>	c2
12	3	2	>	c2
20	11	10	>	c2
25	16	15	>	c2
30	21	20	>	c2

累计上表c1的数据平均 = （2+3+4 ）/3 = 3

累计上表c2的数据平均 = （10+11+12+20+25+30）/6 =  18

数据集（坐标）	到c1距离	到c2距离	距离比较	最后归类
2	1	16	<	c1
3	0	15	<	c1
4	1	14	<	c1
10	7	8	<	c1
11	8	7	>	c2
12	9	6	>	c2
20	17	2	>	c2
25	22	7	>	c2
30	27	12	>	c2

累计上表c1的数据平均 = （2+3+4+10）/4 = 4.75

累计上表c2的数据平均 = （11+12+20+25+30）/5 =  19.6

数据集（坐标）	到c1距离	到c2距离	距离比较	最后归类
2	2.75	17.6	<	c1
3	1.75	16.6	<	c1
4	0.75	15.6	<	c1
10	5.25	9.6	<	c1
11	6.25	8.6	<	c1
12	7.25	7.6	<	c1
20	15.25	0.4	>	c2
25	20.25	5.4	>	c2
30	25.25	15.4	>	c2

最后c1，c2无变化，算法停止，聚类结束！

六、概率模型和统计模型的关系

两者关系

• 不同点：概率模型从理论出发，描述数据。统计模型是从数据触发，获得理论参数。
• 共同点，两者都指向同一组数据，如果高度匹配，能使算法收敛。

七、代码实现

# -*- coding: utf-8 -*-

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import random


data = [
[ 0.697,0.46],[ 0.774,0.376],[ 0.634,0.264],[ 0.608,0.318],
[ 0.556,0.215],[ 0.403,0.237],[ 0.481,0.149],[ 0.437,0.211],
[ 0.666,0.091],[ 0.243,0.267],[ 0.245,0.057],[ 0.343,0.099],
[ 0.639,0.161],[ 0.657,0.198],[ 0.36,0.37],[ 0.593,0.042],
[ 0.719,0.103],[ 0.359,0.188],[ 0.339,0.241],[ 0.282,0.257],
[ 0.784,0.232],[ 0.714,0.346],[ 0.483,0.312],[ 0.478,0.437],
[ 0.525,0.369],[ 0.751,0.489],[ 0.532,0.472],[ 0.473,0.376],
[ 0.725,0.445],[ 0.446,0.459]]



# filepath = open('S:/机器学习-周志华/西瓜数据集4.0.csv')
# data = pd.read_csv(filepath, sep=',')[["密度", "含糖率"]].values.tolist()  # tolist:ndarray转换为list

k = 3  # K值设置，3,4
mean_vectors = random.sample(data, k)  # 初始化均值向量，随机K个
print(mean_vectors)
x0 = list(map(lambda arr: arr[0], mean_vectors))
y0 = list(map(lambda arr: arr[1], mean_vectors))
# 将初试的均值向量用红色的方块表示出
plt.scatter( x0, y0, c='r', marker=',' )


# 计算欧式距离
def Distance(p1, p2):
    sum = 0
    for i, j in zip(p1, p2):
        sum = sum + (i - j) ** 2
    return np.sqrt(sum)


time = 0  # 循环次数
clusters = []  # 聚类初始化
while time < 100:  # 循环次数限制
    clusters = list(map((lambda x: [x]), mean_vectors))
    print(clusters)
    change = 1
    for sample in data:
        dist = []
        for j in mean_vectors:
            dist.append(Distance(j, sample))
        clusters[dist.index(min(dist))].append(sample)

    new_mean_vectors = []
    for c, v in zip(clusters, mean_vectors):
        c_num = len(c)
        c_array = np.array(c)
        v_array = np.array(v)
        new_mean_vector = sum(c_array) / c_num
        # if all(np.divide((new_mean_vector - v), v) < np.array([0.0001, 0.0001])):
        if all(np.true_divide((new_mean_vector - v_array), v_array) < np.array([0.0001, 0.0001])):
            new_mean_vectors.append(v)  # 不变
            change = 0
        else:
            new_mean_vectors.append(new_mean_vector.tolist())  # 更新

    if change == 1:
        mean_vectors = new_mean_vectors
    else:
        break
    time = time + 1

# Show the clustering result

x1 = list(map(lambda arr: arr[0], clusters[0]))
y1 = list(map(lambda arr: arr[1], clusters[0]))
x2 = list(map(lambda arr: arr[0], clusters[1]))
y2 = list(map(lambda arr: arr[1], clusters[1]))
x3 = list(map(lambda arr: arr[0], clusters[2]))
y3 = list(map(lambda arr: arr[1], clusters[2]))
# x4 = list(map(lambda arr: arr[0], clusters[3]))
# y4 = list(map(lambda arr: arr[1], clusters[3]))
# x5 = list(map(lambda arr: arr[0], clusters[4]))
# y5 = list(map(lambda arr: arr[1], clusters[4]))
plt.scatter(x1, y1, c='y', marker='x', label='class1')
plt.scatter(x2, y2, c='g', marker='^', label='class2')
plt.scatter(x3, y3, c='blue', marker='*', label='class3')
# plt.scatter(x4, y4, c='black', marker='+', label='class4')

plt.xlabel('density')
plt.ylabel('sugar_content')
plt.show()

八、 总结

        本讲从收益函数的期望入手，让大家直观理解“期望”就是收益函数的总和，这里注意，期望是两个函数的内积。

        当将收益函数看成是数据的坐标，其几何意义就是数据中心位置。

        加权平均是对期望的另外一个观点。

        期望最大化，是指数据模型和某个概率分布的吻合度，越吻合，期望值越大。

        概率模型和统计模型交互搭配，达到自动迭代运算，并用距离衡量其收敛性。

   

机器学习入门深入浅出聚类算法！如何对王者英雄聚类分析，探索英雄之间的秘密

 Datawhale 

寄语：首先，对聚类算法进行了介绍；然后，解释了EM算法E步、M步的原理；最后，对sklearn参数进行了详解，并对王者荣耀英雄利用EM算法聚类，助力深入理解EM算法。

EM算法(Expectation Maximization Algorithm)，译作最大期望化算法或期望最大算法。它是一种迭代算法，是常见且经典的聚类算法之一，用于含有隐变量（hidden variable）的概率参数模型的最大似然估计或极大后验概率估计。

对聚类算法、EM算法的原理及其实践进行详细的讲解之前。首先来看一张EM算法的聚类图，有个大致直观地了解。

学习框架

数据集及聚类分析代码后台回复 王者荣耀 获取

聚类算法

先来一段西瓜书里面的定义： 在“无监督学习”中，训练样本的标记信息是未知的，目标是通过对无标记训练样本的学习来揭示数据的内在性质及规律，为进一步的数据分析 提供基础，此类学习任务中研究最多、应用最广的是“聚类”（clustering）。

总结一下关键词：标记信息未知、学习、内在性质及规律、聚类。对，还有一个：无监督学习，无监督算法大多都可以用上面的关键词来描述。

以聚类算法为例，其目的是对一批未知标记的数据通过某种方式进行聚类，使其能够有效的分成若干个类别，每一个类别里面的数据特征都类似，不同类别的数据差异性较大。

举个简单的例子：在中国的乡村有这样一个现象，一个村子的姓氏大多相同，不同村子有不同的姓氏。那如果现在把王家村、李家村、赵家村的所有人都聚集在一起，前提是不知道他们是哪个村子的，如何对他们进行聚类？

• 特性①：姓名，通过姓氏进行聚类，最终聚成三类：王+李+赵+其它

• 特性②：性别，通过性别进行聚类，最终聚成两类：男+女

• 特性③：年龄，通过年龄进行聚类，最终聚成三类：儿童+青年+老年

• 特性④：价值，通过价值进行聚类，最终聚成三类：高收入+中收入+低收入

• 特性⑤：属性，通过属性进行聚类，最终聚成三类：村领导+村干部+村民

• …

上面的姓氏、性别、年龄、价值和属性等都是村民的内在性质（人物特征），这样的内在性质有很多，也就决定了聚类的标准并不唯一。

ok，想必大家已经明白了什么是聚类，通过上面的例子我们总结一下。

1. 何为聚类

聚类：将数据集中的样本划分为若干个不相交的子集，每个子集内部的样本之间具有相同的性质，不同子集之间差异性较大。通常情况下，我们会将子集称之为“簇”(cluster)

2. 如何聚类

聚类的本质是将具有相似特征的样本划分在一个簇里面，根据聚类算法的不同，聚类的实现过程也不尽相同。

例如，聚类算法中k-means是基于均值的聚类，DBSCAN是基于密度的聚类，AGNES是基于层次的聚类，可以针对不同的样本集使用不同算法进行聚类。

3. 评估聚类

聚类性能的评估比较麻烦，主要有两个原因：

• 样本集不存在已标记的类别数据，无法直接计算聚类算法的准确率。

• 若存在标记类别数据，无法直接通过预测前后类别之间的对应关系进行性能评估

针对上面的问题，可以大致分为两种，一种是存在已经确定的标记类别数据（类似于分类数据集），一种是完全没有标记的类别数据。

有标记类别数据的评估：当前的样本数据有标记类别数据C1和预测后的标记类别数据C2，但是无法直观的通过C1、C2去计算聚类错误率。

这个时候，可以通过条件熵去分析，可以认识到两个指标，分别是齐次性和完整性。通过这两个指标可以评估带有类别标记样本的聚类性能。

其中齐次性表示一个聚类元素只由一种类别的元素组成；完整性则表示给定的已标记的类别 ，全部分配到一个聚类里。

没有标记的类别数据的评估：大多应用于聚类算法的数据都是无标记的，因为既然数据都有标记了，直接用分类算法不香吗？

有一个指标叫做轮廓系数，它可以在不需要已标记数据集的前提下，对聚类算法的性能进行评估。

轮廓算法由以下两个指标构成：

• a：一个样本与其所在相同聚类的平均距离

• b：一个样本与其距离最近的下一个聚类里的点的平均距离

则针对这个样本，其轮廓系数s的值为：

针对一个数据集，其轮过系数s 为其所有样本的轮廓系数的平均值。轮廓系数的数值介于[-1,1]之间，-1表示完全错误的聚类，1表示完美的聚类，0表示聚类重叠。

EM原理

EM的英文全称是：Expectation Maximization，所以EM算法也叫最大期望算法。

学习EM之前，希望你已经理解了什么是极 大似然估计，不了解的点这个：

1. 极大似然估计

先说一下极大似然估计：已知某个随机样本满足某种概率分布，且某个参数能使这个样本出现的概率最大，我们把这个参数作为估计的真实值叫做最大似然估计。也就是求解出现样本结果的最佳参数θ。

所以极大似然估计需要面临的概率分布只能有一个，但是要是不止一个呢？看个例子：

假设现在有两枚硬币A和B，随机抛掷后正面朝上概率分别为P_A，P_B。 为了估计这两个概率，需要每次取一枚硬币，连掷10下并记录下结果，结果如下：

ok，根据以上分布结果，可以轻松算出：

似乎很简单，但是你要知道每次我们都知道了是抛出哪个硬币。换句话说，我们已经提前知道了每次选择硬币抛出的样本分布。如果，我们不知道呢？那会是什么样的？上面的例子变成了这种：

Z{A,B}表示每次选择A、B两个硬币中一个，但是我们现在已经不知道是哪一个了，目标还是估计A、B两个硬币正面朝上的概率。当我们知道每一次抛出的硬币是A还是B，才能利用最大似然估计对A、B正面朝上的概率进行估计。

当我们知道A、B正面朝上的概率，才会知道抛出的硬币最有可能是A还是B。这样来看，这个问题就有趣了，我们并不知道是鸡生蛋还是蛋生鸡，怎么判断是先有鸡还是先有蛋？考虑这个问题不妨再来看下面的例子。

2. 分菜问题

大厨炒好了一锅菜，需要把锅里的菜平均分配到两个碟子里，大厨应该怎么办呢？

如果是浸淫已久的大厨，可能随便小手一抖就平均到两个碟子里了，但是大多数都是伪大厨，手上功夫还不到位，他们应该怎么做？

伪大厨可能需要多试几次才行，比如先随便分在两个碟子里，然后把菜多碟子的菜往另一个碟子匀匀，然后再匀匀，再匀匀…一直到两个碟子中菜的量大致一样。

ok，那我们这个问题呢？要不也试试伪大厨的方法？

3. 模仿分菜

首先，我们模仿伪大厨的方法，随便设置A、B硬币正面朝上的概率P_A=0.5、P_B=0.9。如果第一枚硬币是A硬币，那么它正面朝上的概率是：

如果第一枚硬币是B硬币，那么它正面朝上的概率是：

同样的计算方法，其他几枚硬币的概率分别是：

按照最大似然法则，每一次取概率最大的作为我们的结果，则硬币的顺序依次是：AABBA。 这个也就是上面我 们假设的Z值，那么根据这个Z值我们可以轻松的算出：

算出的P_A、P_B的值和我们假设的不一致，这时候说明伪大厨还没有把菜分均匀。看来伪大厨还需要再分一次，这个时候A、B正面朝上的值由第一次的：

更新成现在的：

重复上面的步骤，最终A、B正面朝上的概率不再发生变化，且会逐渐逼近一个值，这就是EM算法的工作原理。

4. 模仿的升级

虽然说我们一直在模仿大厨的操作，但是我们也想要超越他成为更厉害的大厨。

在上面的过程中，我们发现直接每一次选择最大概率的结果作为硬币的选择有点过于绝对，因为从计算结果来看另一枚硬币也是有概率的，只是概率偏小一点。这样的话，我们在第一轮中可以这样计算：

如果第一枚硬币是A硬币，那么它正面朝上的概率是：

同理可以求出第2轮到底5轮的概率值

此时，我们可以根据最大似然估计求出的概率，分别算出AB正反面的期望值：

例如：第一轮中，0.994的概率为A，抛10次，正面朝上的概率为0.994*5=9.94，同理反正为0.06。 同理可以求出第2轮到第5轮的期望值：

此时根据期望值我们可以轻松的算出：

可以看出，相比上一种方法，这种方法的收敛会更快些，更加逼近我们的目标值。这个过程就是EM算法的实现过程。

5. EM工作原理

上面的投掷硬币属于A硬币还是B硬币我们称之为隐含参数，A硬币和B硬币的分布参数我们称之为模型参数。

EM 算法解决这个的思路是使用启发式的迭代方法，既然我们无法直接求出模型分布参数，那么我们可以先猜想隐含参数（EM算法的E步），接着基于观察数据和猜测的隐含参数一起来极大化似然估计，求解我们的模型参数（EM算法的M步)。

由于我们之前的隐含参数是猜测的，所以此时得到的模型参数一般还不是我们想要的结果。我们基于当前得到的模型参数，继续猜测隐含参数（EM算法的E步），然后继续极大化似然估计，求解我们的模型参数（EM算法的M步)。

以此类推，不断的迭代下去，直到模型分布参数基本无变化，算法收敛，找到合适的模型参数。

画了一个图，感受一下：

EM聚类

EM算法在聚类的时候也是要先估计一个隐状态，这个隐状态也就是我们的样本标签。

有了样本标签之后，就可以将原来的无监督学习转换为监督学习，然后通过极大似然估计法求解出模型最优参数。

需要解释一点的是，在整个过程中，隐状态的估计需要用到EM算法。

硬聚类or软聚类

k-means算法是通过距离来聚类的，因为距离是确定的，所以就导致每个样本只能归为一类，这叫做硬聚类。

而EM算法在聚类的过程中，每个样本都有一定的概率和每个聚类有关，这叫做软聚类。

通常，我们可以假设样本符合高斯分布，因为每个高斯分布都属于这个模型的组成部分，要分成K个簇就相当于是K个组成部分。

这样我们可以先初始化每个组成部分的高斯分布的参数，然后再来看每个样本是属于哪个组成部分。这就是E步骤；再通过得到的这些隐含变量结果，反过来求每个组成部分高斯分布的参数，即 M 步骤。

反复EM步骤，直到每个组成部分的高斯分布参数不变为止，这样也就相当于将样本按照高斯模型进行了EM聚类。

EM 算法相当于一个框架，我们可以采用不同的模型来进行聚类，比如 GMM（高斯混合模型）、 HMM（隐马尔科夫模型）来进行聚类。

GMM通过概率密度来进行聚类，聚成的类符合高斯分布（正态分布）。HMM用到了马尔可夫过程，通过状态转移矩阵来计算状态转移的概率。

项目实战

1. 准备工作

如何创建高斯聚类呢，我们需要先了解一下高斯聚类的参数。在sklearn 中，高斯聚类可以这样创建：

# 创建高斯聚类模型gmm = GaussianMixture(n_components=1, covariance_type='full', max_iter=100)

解释一下主要的参数：

• n_components：即高斯混合模型的个数，也就是我们要聚类的个数，默认值为 1。

• covariance_type：代表协方差类型。一个高斯混合模型的分布是由均值向量和协方差矩阵决定的，所以协方差的类型也代表了不同的高斯混合模型的特征。

• max_iter：代表最大迭代次数，EM 算法是由 E 步和 M 步迭代求得最终的模型参数，这里可以指定最大迭代次数，默认值为 100。

其中协方差类型covariance_type又四种取值，分别是：

• covariance_type=full，代表完全协方差，也就是元素都不为 0；

• covariance_type=tied，代表相同的完全协方差；

• covariance_type=diag，代表对角协方差，也就是对角不为 0，其余为 0；

• covariance_type=spherical，代表球面协方差，非对角为 0，对角完全相同，呈现球面的特性。

需要注意的是，聚类的个数往往是由业务决定的，比如对用户收入进行聚类，可以分为：高收入群体、中收入群体、低收入群体，根据用户价值进行聚类，可以分为：高价值用户、中价值用户、低价值用户、无价值用户等等。

当然如果你无法确定聚类的个数，可以通过设置不同的聚类个数进而选择具有最优效果的模型。

2. 了解数据

本次实战我们的数据是王者荣耀的英雄属性数据，通过对69个英雄的22个属性数据，其中包括英雄的最大生命、生命成长、最大发力、最高物攻等等，通过每个英雄之间的特征，对69个英雄进行“人以群分，物以类聚”。

感兴趣的同学可以尝试一下最终的结果能否应用于实际游戏中。ok，先来看看我们本次数据的整体描述：

df_ori.head(5)

再来看看各个英雄属性的整体情况：

"""数据的EDA操作""""""1.数据整体描述"""df.data=df_ori.copy()df.data.drop('英雄'，axis=1,inplace=True)#次要定位存在空值df.data.info()

一共22个英雄属性（不包括姓名），其中次要定位存在空值，且空值较多。再来看看数值型数据的整体分布情况：

df_data.describe()

数据分布没有什么异常，但是应该需要考虑进行标准化，这个后面再说。最大攻速字段应该是数值型的，我们需要对其进行处理：

df_ori.head(5)

另外，次要定位属性缺失值太多，而且没有有效的填充方法，直接删掉它

# 最大攻速为百分比需要替换成数值df_data['最大攻速'] = df_data['最大攻速'].apply(lambda str: str.replace('%', ''))# 次要定位数据无法填充，且已存在主要定位，直接删除该字段df_data.drop('次要定位', axis=1, inplace=True)

3. 数据探索

一共只有69数据，但是却有22个属性，是否存在属性重复的情况呢？我们知道在建模过程中，重复的属性对最终结果不会产生影响。所以我们可以通过关联度分析，看一下数据之间的关联度情况，这种方式在前面的实战种很多次都用到过。

可以看到，用红框标出的，都是属性之间相互关联度特别大的，对于这些我们只需要保留其中的一种属性即可。通过筛选，我们最终确定了13个英雄属性

"""进行特征选择"""features=df_data.columns.values.tolist()duplicates_features=['生命成长','最大法力','法力成长','物攻成长','物防成长','每5秒回血成长','最大每5秒回血成长','每5秒回蓝成长']features=featuresfor feature in duplicates_features:    features.remove(feature)df_data=df_data[features]df_data.head()

再来看英雄属性：攻击范围和主要定位，是离散型特征，直接对其进行特征量化

"""通过标签编码实现特征量化"""for feature in ['攻击范围', '主要定位']: le = preprocessing.LabelEncoder() le.fit(df_data[feature])  df_data[feature] = le.transform(df_data[feature])

最后就是数据的规范化，直接通过Z-Score进行数据规范

"""采用Z-Score规范化数据，保证每个特征维度的数据均值为0，方差为1"""stas = StandardScaler()df_data = stas.fit_transform(df_data)

4. 建模

选用我们前面提到的GMM进行建模

# 构造GMM聚类gmm = GaussianMixture(n_components=30, covariance_type='full')gmm.fit(df_data)
# 训练数据prediction = gmm.predict(df_data)

最终的模型聚类结果是这样的：

#将分组结果输出到CSV文件中df_ori.insert(0,'分组',prediction)df_ori.sort_values('分组').head(15)

5. 总结

上面的图中放了前20的英雄，组号相同的英雄表示属性相近，感兴趣的同学不妨在游戏中试试？

另外，聚类算法属于无监督的学习方式，我们并没有实际的结果可以进行对比来区别模型的准确率。这里我们试着用轮廓系数进行模型的评分。

"""根据轮廓系数计算模型得分"""from sklearn.metrics import silhouette_scorescore=silhouette_score(df_data,prediction,metric='euclidean')score

最后得分0.246，也不是很高，说明聚类的效果不是特别好，应该还是英雄属性的原因，例如，通过主要定位就可以对英雄就行聚类，或者通过攻击范围进行聚类，但是这两个属性和其他属性的结合，有时候并非是最好的，对游戏理解比较深刻的同学可以考虑一下优化方法。

王者荣耀英雄数据集及聚类实践代码：https://github.com/double-point/machine_learning/tree/master/EM

    
      
      
    

     
       
       
     
 
      
 
       
 
        
 
         
 
          
 
           
 
            
往期精彩回顾
 
           
 
          
 
         
 
        
 
       
 
      
 
     
    
      
      
    
    
      
      
    

     
       
       
     
 
      
 
       
 
        
 
       
 
       
 
        
 
         
 
        
 
        
 
         
 
        
 
       
 
       
 
        
 
       
 
      
 
      
 
       
 
       
 
       
 
       
 
       
 
       
 
      
 
      
本站qq群1003271085，加入微信群请回复“加群”
 
      
获取一折本站知识星球优惠券，复制链接直接打开：
 
      
https://t.zsxq.com/yFQV7am
 
      
喜欢文章，点个在看