机器学习-KMeans

一、 K-均值聚类算法

聚类是一种无监督学习,它将相似的对象归到同一个簇中。簇内对象越相似,聚类效果越好。K-Means算法就是其中的一种聚类算法。其实现代码如下:

#1.KMeans均值聚类算法
#2.伪代码
"""
       创建k个点作为起始质心(随机选择)
       当任意一个点的簇分配结果发生改变时
           对数据中的每个数据点
               对每个质心
                   计算质心与数据点之间的距离
                将数据点分配到距其最近的簇
            对每一个簇,计算簇中所有点的均值并将均值作为质心
"""
def KMeans(dataSet,k,distMeas=disEclud,createCent=randCent):
    """
    :param dataSet:      数据集
    :param k:            质心数目
    :param distMeas:     计算欧式距离函数
    :param createCent:   创建随机质心函数
    :return:
    """
    m=shape(dataSet)[0]
    clusterAssment=mat(zeros((m,2)))          #创建矩阵来存储每个点的簇分配结果,第一列为簇的索引值,第二例为误差
    centroids=createCent(dataSet,k)
    clusterChanged=True
    while clusterChanged:
        clusterChanged=False
        for i in range(m):
            minDist=inf;
            minIndex=-1
            for j in range(k):
                disJI=distMeas(centroids[j,:],dataSet[i,:]) #寻找质心,通过对每个点遍历所有质心并计算点到每个质心的距离来完成
                if disJI<minDist:
                    minDist=disJI
                    minIndex=j
            if clusterAssment[i,0]!=minIndex:               #如果任一点的簇分配结果发生改变,则更新标志位
                clusterChanged=True
            clusterAssment[i,:]=minIndex,minDist**2
        print(centroids)

        for cent in range(k):                               #遍历所有质心,并更新他们的取值
            ptsInClus=dataSet[nonzero(clusterAssment[:,0].A==cent)[0]]
            centroids[cent,:]=mean(ptsInClus,axis=0)
    return centroids,clusterAssment

K-均值算法收敛但聚类效果较差的原因是:K-均值算法收敛到了局部最小,而非全局最小。

二、 二分K-均值聚类算法

为了克服K-均值算法收敛于局部最小,可以使用二分K-均值算法。该算法首先将所有点作为一个簇,然后将该簇一分为二,之后选择其中的一个簇继续进行划分,选择哪一簇划分取决于对其划分是否可以最大程度降低SSE(误差平方和)的值。上述过程基于SSE划分过程不断重复,直到得到用户指定的簇数目位置。实现代码如下:

#1.二分KMeans算法
#2.伪代码
"""
      将所有点看成一个簇
      当簇的数目小于k时
          对于每一个簇
              计算总误差
              在给定的簇上面进行KMeans聚类(k=2)
              计算将该簇一分为二之后的总误差
          选择使得误差最小的那个簇进行划分操作        
"""
def biKMeans(dataSet,k,distMeas=disEclud):
    m=shape(dataSet)[0]
    clusterAssment=mat(zeros((m,2)))             #创建矩阵用于保存分配结果以及平方误差
    centroid0=mean(dataSet,axis=0).tolist()[0]   #将所有点看成一个簇,求得其平均值的中心
    centList=[centroid0]                         #中心值列表,最开始只有一个簇,所以只有一个中心
    for j in range(m):
        clusterAssment[j,1]=distMeas(mat(centroid0),dataSet[j,:])**2  #求得一个簇时,循环求得所有点到中心点的误差,并放入矩阵中
    while(len(centList)<k):                      #判断是否分到想要的簇数
        lowestSSE=inf                            #初始最小误差平方和SSE为正无穷
        for i in range(len(centList)):           #循环每个簇
            ptsInCurrCluster=dataSet[nonzero(clusterAssment[:,0].A==i)[0],:]  #分类结果等于i数据集的提取出来
            centroidMat,splitClustAss=KMeans(ptsInCurrCluster,2,distMeas)     #使用KMeans进行二分类
            sseSplit=sum(splitClustAss[:,1])                                  #分类之后的总误差
            sseNotSplit=sum(clusterAssment[nonzero(clusterAssment[:,0].A!=i)[0],1])#分类之前不等于i的分类结果的总误差
            print("sseSplit,and notSplit:",sseSplit,sseNotSplit)
            if(sseSplit+sseNotSplit)<lowestSSE:                               #分类之后的总误差+分类之前的总误差小于最小误差
                bestCentToSplit=i                                             #分类i
                bestNewCents=centroidMat                                      #新分类点的中心坐标
                bestClusAss=splitClustAss.copy()                              #分类之后的结果以及平方误差
                bestClustSSE=sseNotSplit+sseSplit                             #分类之后的平方误差

        #更新簇的分配结果
        bestClusAss[nonzero(bestClusAss[:,0].A==1)[0],0]=len(centList)
        bestClusAss[nonzero(bestClusAss[:,0].A==0)[0],0]=bestCentToSplit

        print("the bestCentToSplit is:",bestCentToSplit)
        print("the len of bestClustAss is :",len(bestClusAss))
        centList[bestCentToSplit]=bestNewCents[0,:]                           #将新分配的坐标中心替换
        centList.append(bestNewCents[1,:])                                    #将第二个坐标中心加入cenList中
        clusterAssment[nonzero(clusterAssment[:,0].A==bestCentToSplit)[0],:]=bestClusAss
    return  centList,clusterAssment

三、 完整的实现代码


"""
1.机器学习-KMeans
2.姓名:pcb
3.日期:2019.01.14
"""

from numpy import *
import matplotlib.pyplot as plt

#加载数据
def loadDataSet(filename):
    dataMat=[]
    fr=open(filename)
    for line in fr.readlines():
        curLine=line.strip().split('\t')
        #fltLine=map(float,currLine)
        fltLine = []
        for i in curLine:
            fltLine.append(float(i))
        dataMat.append(fltLine)
    return dataMat

#计算两个向量的欧式距离
def disEclud(vecA,vecB):
    return sqrt(sum(power(vecA-vecB,2)))

#给定数据集,构建一个包含k个随机质心的集合
def randCent(dataSet,k):
    n=shape(dataSet)[1]
    centroids=mat(zeros((k,n)))
    for j in range(n):
        minJ=min(dataSet[:,j])
        rangeJ=float(max(dataSet[:,j])-minJ)
        centroids[:,j]=minJ+rangeJ*random.rand(k,1)
    return centroids

#1.KMeans均值聚类算法
#2.伪代码
"""
       创建k个点作为起始质心(随机选择)
       当任意一个点的簇分配结果发生改变时
           对数据中的每个数据点
               对每个质心
                   计算质心与数据点之间的距离
                将数据点分配到距其最近的簇
            对每一个簇,计算簇中所有点的均值并将均值作为质心
"""
def KMeans(dataSet,k,distMeas=disEclud,createCent=randCent):
    """
    :param dataSet:      数据集
    :param k:            质心数目
    :param distMeas:     计算欧式距离函数
    :param createCent:   创建随机质心函数
    :return:
    """
    m=shape(dataSet)[0]
    clusterAssment=mat(zeros((m,2)))          #创建矩阵来存储每个点的簇分配结果,第一列为簇的索引值,第二例为误差
    centroids=createCent(dataSet,k)
    clusterChanged=True
    while clusterChanged:
        clusterChanged=False
        for i in range(m):
            minDist=inf;
            minIndex=-1
            for j in range(k):
                disJI=distMeas(centroids[j,:],dataSet[i,:]) #寻找质心,通过对每个点遍历所有质心并计算点到每个质心的距离来完成
                if disJI<minDist:
                    minDist=disJI
                    minIndex=j
            if clusterAssment[i,0]!=minIndex:               #如果任一点的簇分配结果发生改变,则更新标志位
                clusterChanged=True
            clusterAssment[i,:]=minIndex,minDist**2
        print(centroids)

        for cent in range(k):                               #遍历所有质心,并更新他们的取值
            ptsInClus=dataSet[nonzero(clusterAssment[:,0].A==cent)[0]]
            centroids[cent,:]=mean(ptsInClus,axis=0)
    return centroids,clusterAssment

#1.二分KMeans算法
#2.伪代码
"""
      将所有点看成一个簇
      当簇的数目小于k时
          对于每一个簇
              计算总误差
              在给定的簇上面进行KMeans聚类(k=2)
              计算将该簇一分为二之后的总误差
          选择使得误差最小的那个簇进行划分操作        
"""
def biKMeans(dataSet,k,distMeas=disEclud):
    m=shape(dataSet)[0]
    clusterAssment=mat(zeros((m,2)))             #创建矩阵用于保存分配结果以及平方误差
    centroid0=mean(dataSet,axis=0).tolist()[0]   #将所有点看成一个簇,求得其平均值的中心
    centList=[centroid0]                         #中心值列表,最开始只有一个簇,所以只有一个中心
    for j in range(m):
        clusterAssment[j,1]=distMeas(mat(centroid0),dataSet[j,:])**2  #求得一个簇时,循环求得所有点到中心点的误差,并放入矩阵中
    while(len(centList)<k):                      #判断是否分到想要的簇数
        lowestSSE=inf                            #初始最小误差平方和SSE为正无穷
        for i in range(len(centList)):           #循环每个簇
            ptsInCurrCluster=dataSet[nonzero(clusterAssment[:,0].A==i)[0],:]  #分类结果等于i数据集的提取出来
            centroidMat,splitClustAss=KMeans(ptsInCurrCluster,2,distMeas)     #使用KMeans进行二分类
            sseSplit=sum(splitClustAss[:,1])                                  #分类之后的总误差
            sseNotSplit=sum(clusterAssment[nonzero(clusterAssment[:,0].A!=i)[0],1])#分类之前不等于i的分类结果的总误差
            print("sseSplit,and notSplit:",sseSplit,sseNotSplit)
            if(sseSplit+sseNotSplit)<lowestSSE:                               #分类之后的总误差+分类之前的总误差小于最小误差
                bestCentToSplit=i                                             #分类i
                bestNewCents=centroidMat                                      #新分类点的中心坐标
                bestClusAss=splitClustAss.copy()                              #分类之后的结果以及平方误差
                bestClustSSE=sseNotSplit+sseSplit                             #分类之后的平方误差

        #更新簇的分配结果
        bestClusAss[nonzero(bestClusAss[:,0].A==1)[0],0]=len(centList)
        bestClusAss[nonzero(bestClusAss[:,0].A==0)[0],0]=bestCentToSplit

        print("the bestCentToSplit is:",bestCentToSplit)
        print("the len of bestClustAss is :",len(bestClusAss))
        centList[bestCentToSplit]=bestNewCents[0,:]                           #将新分配的坐标中心替换
        centList.append(bestNewCents[1,:])                                    #将第二个坐标中心加入cenList中
        clusterAssment[nonzero(clusterAssment[:,0].A==bestCentToSplit)[0],:]=bestClusAss
    return  centList,clusterAssment

def main():
# #1.--------KMeans----------------------------------
#     dataMat=mat(loadDataSet('testSet.txt'))
#     myCentroids,clustAssing=KMeans(dataMat,4)
# #--------------------------------------------------

#2----------二分KMeans算法---------------------------
    dataMat3=mat(loadDataSet('testSet3.txt'))
    centList,myNewAssments=biKMeans(dataMat3,3)
    print (centList)
#---------------------------------------------------
if __name__=='__main__':
    main()
### CEEMDAN KMeans 算法实现及应用 #### CEEMDAN 和 KMeans 的结合原理 CEEMDAN (Complete Ensemble Empirical Mode Decomposition with Adaptive Noise) 是一种信号处理技术,能够有效地将复杂的时间序列分解成多个固有模式函数(IMFs)[^1]。K-means 聚类算法则是一种无监督机器学习方法,旨在将数据集划分成若干个簇,使每个数据点到其所在簇中心的距离最小化。 当这两种技术结合起来时,首先利用 CEEMDAN 对原始时间序列进行多尺度分解,获得一系列 IMF 分量;接着提取各 IMF 组件的能量、熵、频率等特征参数,并以此作为输入喂给 K-means 算法来进行分类。这种组合方式有助于识别并分离出不同性质的波动成分,从而为进一步的数据分析提供了便利条件。 #### 实现过程中的关键技术要点 为了更好地理解如何具体实施这一流程,在此给出一个简化版 Python 伪代码框架: ```python import numpy as np from pyhht.emd import EMD from sklearn.cluster import KMeans def ceemdan_kmeans(data, num_imfs=5): """ 使用 CEEMDAN 和 KMeans 处理一维时间序列 参数: data -- 输入的一维数组形式的时间序列 num_imfs -- 需要计算的最大 IMFs 数目,默认为 5 返回值: labels -- 各 IMF 所属类别标签列表 centers -- 各类别对应的质心向量矩阵 """ # Step 1: Perform CEEMDAN decomposition on the input signal emd = EMD() imfs = emd.ceemdan(data) # Extract features from each IMF component feature_vectors = [] for i in range(num_imfs): energy = sum(imfs[i]**2) entropy = -np.sum((imfs[i]/energy)*np.log(abs(imfs[i])/energy)) freq = abs(np.fft.rfft(imfs[i])).mean() feature_vector = [energy, entropy, freq] feature_vectors.append(feature_vector) # Convert list of lists into a NumPy array X = np.array(feature_vectors).reshape(-1, len(feature_vectors[0])) # Apply k-means clustering algorithm to classify IMFs based on their extracted features model = KMeans(n_clusters=num_imfs//2+1, random_state=42) labels = model.fit_predict(X) centers = model.cluster_centers_ return labels, centers ``` 上述代码展示了怎样运用 `pyhht` 库执行 CEEMDAN 分解操作以及调用 scikit-learn 中内置的 KMeans 函数完成聚类任务。需要注意的是这只是一个非常基础的例子,实际应用场景下可能还需要考虑更多细节因素,比如预处理步骤的选择、模型超参数调整等问题。 #### 应用实例说明 在一个典型的应用场景中,假设有一个金融市场的股票价格走势记录下来形成了一条连续变化的时间序列。通过 CEEMDAN-KMeans 方法对其进行分析后发现某些特定类型的波动(如短期快速涨跌)往往集中在某一类别的IMF之中,而长期缓慢上升的趋势则主要体现在另一些类别里。因此可以根据这样的规律采取针对性策略——对于前者可设置止损位防止损失扩大;而对于后者,则可以在适当时候增加仓位获取更大收益。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值