20150913K-means聚类

1.聚类的思想:

将一个有N个对象的数据集,构造成k(k<=n)个划分,每个划分代表一个簇。使得每个簇包含一个对象,每个对象有且仅属于一个簇。
对于给定的k,算法首先给出一个初始的划分方法,以后通过反复迭代的方法改变划分,使得每一次改进之后的划分方案都较前一次更好

2.K-means聚类

2.1K-means聚类的思想

K-means算法使用广泛,有时候也作为其他聚类算法的基础。
算法首先随机选择k个对象,每个对象初始地代表了一个簇的平均值和中心。对剩余的每个对象根据其与各个簇中心的距离,将它赋给最近的簇。然后重新计算每个簇的平均值。这个过程不断重复,直到准则函数(常常使用最小平方误差)收敛。
其损失函数表达式为\(\sum_{i=1}^k \sum_{x_j \in S_j} \left(x_j - \mu_j \right)^2\)

2.2K-means聚类的算法步骤

输入:n个样本数据,分别为$x_1,x_2,\cdots,x_n$
1. 随机选择k个聚类中心,$\mu_1,\mu_2,\cdots,\mu_k \in \mathbb{R}^{n}$
2. 针对剩余的样本数据,将其类别标签设为距离其最近的聚类中心的标签。
3. 将每个聚类中心的值更新为与该类所有样本的平均值。
4. 重复以上步骤,直到聚类中心的变化小于规定的阈值即可。

2.3K-means聚类代码实现

2.3.1随机构造二维数据集

# -*- coding:utf-8 -*-
import matplotlib.pyplot as plt
from numpy import *
import random
flag = ['b*','g+','rs','sb', 'db', '<b', 'pb']

def create():
    num = 100
    data = [[],[],[]]
    dataCenter = [(2,3),(2.5,3),(2,2.5)]
    for i in xrange(len(data)):
        for j in xrange(num):
            data[i].append((dataCenter[i][0]+random.uniform(-1,1)**2*random.uniform(-1,1), dataCenter[i][1]+random.uniform(-1,1)**2*random.uniform(-1,1)))
    ##draw picture
    global flag
    for i in xrange(len(data)):
        for j in data[i]:
            plt.plot(j[0],j[1],flag[i])
    plt.show()
    return data[0]+data[1]+data[2]

聚类前

2.3.2K-means聚类python实现

# -*- coding:utf-8 -*-
import matplotlib.pyplot as plt
from numpy import *
import random
##计算两个点之间的距离,这里采用的是欧式距离,关于距离的选择,看场景。
def distEclud(vecA, vecB):
    return sqrt(sum(power(vecA-vecB,2)))

##初始化选择k个质心,这里选用的是从n个点中随机选出k个质心。
def initCenter(data, k=2):
    n = shape(data)[1]
    centers = mat(zeros((k,n)))
    for i in xrange(k):
        index = int(random.uniform(0,len(data)))
        centers[i] = data[index]
    return centers

##计算上次和本次质心的距离偏差
def deltaCenter(centers, centersNext):
    return sqrt(sum(power(centers-centersNext,2)))

def KMeans(data,k=2):
    centers = initCenter(data, k)
    centersNext = mat(zeros((k,shape(data)[1])))
    dataNum = shape(data)[0]
    clusterRes = mat(zeros((dataNum,2)))
    eps = 0.01
    delta = inf
    freq = 10
    while eps < delta and freq>0:
        freq -= 1
        ##针对每个点划分类别
        for i in xrange(dataNum):
            mindist = inf
            for j in xrange(k):
                dist = distEclud(centers[j],data[i])
                if dist<mindist:
                    mindist = dist
                    minIndex = j
            clusterRes[i] = [minIndex,mindist]
        ##recalc the center of cluster
        for j in xrange(k):
            clusterData = data[nonzero(clusterRes[:,0].A==j)[0]]
            centers[j] = mean(clusterData, axis=0)
        ##计算上一次聚类中心和这一次的变动
        delta = deltaCenter(centers,centersNext)
    ##draw results
    # global flag
    # for i in xrange(k):
    #     clusterData = data[nonzero(clusterRes[:,0].A==i)[0]]
    #     for j in clusterData:
    #         plt.plot(j[0,0],j[0,1],flag[i])
    # plt.show()
    return centers,clusterRes

if __name__ == '__main__':
    data = create()
    KMeans(mat(data),3)

聚类后
3个聚类中心依次为:[1.95408899 2.44211387] [ 1.97371672 3.04483904][ 2.59031316 2.9787235 ]

2.4K-means聚类缺点

初值敏感
对噪声敏感
不适于发现非凸面形状的簇或大小差别很大的簇
无法保证收敛到全局最优
有可能会出现某个聚类中心没有任何样本

3.二分K-means

3.1二分K-means算法步骤

由于K-means算法有时候容易收敛到局部最小值,因此就提出了二分K-means。
该算法是将所有点作为一个簇,然后将该簇一分为二,之后选择其中一个簇进行划分,选择哪一个簇进行划分取决于对其划分是否可以最大程度降低SSE(误差平方和)的值,直到得到指定的簇数目为止。
聚类的SSE能够衡量聚类的性能,该值越小表示数据点越接近于它们的质心,聚类效果越好。所以

将所有数据看成一个簇,对该簇进行二分K-means
当簇的数目小于k时
    对每一个簇
         进行K-means划分,其中K=2
         计算划分后的总误差
    选择总误差最小的那个簇进行划分

3.2二分K-means代码实现

# -*- coding:utf-8 -*-
########################################
# kmeans: k-means cluster
# Author : xuke
# Date   : 2015-09-14
########################################
import matplotlib.pyplot as plt
from numpy import *
import random
flag = ['b*','g+','rs','sb', 'db', '<b', 'pb']

def biKmeans(data,k=2):
    dataNum = shape(data)[0]
    clusterRes = mat(zeros((dataNum,2)))
    center0 = mean(data,axis=0).tolist()[0]
    centers = [center0]
    ##计算一簇的簇心
    for i in xrange(dataNum):
        clusterRes[i,1] = distEclud(mat(center0), data[i])
    while (len(centers) < k):
        minSSE = inf
        ##遍历每个簇,划分每个簇,求得每个簇的SSE
        for i in xrange(len(centers)):
            dataNow = data[nonzero(clusterRes[:,0].A==i)[0],:]
            centerTemp, clusterResTemp = KMeans(dataNow,2)
            dataSSE = sum(clusterResTemp[:,1])
            dataNoSSE = sum(clusterRes[nonzero(clusterRes[:,0].A!=i)[0],1])
            if dataSSE + dataNoSSE < minSSE:
                bestCluster = i
                bestCenter = centerTemp
                bestClusterResTemp = clusterResTemp.copy()
                minSSE = dataSSE + dataNoSSE
        ##为选择出的最佳划分簇,打上类别,假设第2簇需要划分,且现在共有4簇,则第2簇打上2,4label。
        bestClusterResTemp[nonzero(bestClusterResTemp[:,0].A == 1)[0],0] = len(centers)
        bestClusterResTemp[nonzero(bestClusterResTemp[:,0].A == 0)[0],0] = bestCluster
        ##将最佳质心append进centers这个list
        centers[bestCluster] = bestCenter[0,:].tolist()[0]
        centers.append(bestCenter[1,:].tolist()[0])
        clusterRes[nonzero(clusterRes[:,0].A == bestCluster)[0],:] = bestClusterResTemp
    ##draw results
    global flag
    for i in xrange(k):
        clusterData = data[nonzero(clusterRes[:,0].A==i)[0]]
        for j in clusterData:
            plt.plot(j[0,0],j[0,1],flag[i])
    plt.show()
    return mat(centers), clusterRes

if __name__ == '__main__':
    data = create()
    # print KMeans(mat(data), 4)[0]
    print biKmeans(mat(data), 3)[0]
posted @ 2016-07-24 10:57  ericxk  阅读(231)  评论(0编辑  收藏  举报