数据挖掘算法之k-means算法

#include<iostream>
#include<fstream>
#include<vector>
#include<random>
#include<time.h>
#include<string.h>
using namespace std;
const int maxNum=0x1<<30;
const int repeatMax=100;//控制迭代的上限，这里主要从效率的角度来考虑。一般来说迭代50--100次就能达到很好的效果
const int AttributeCount=2;//数据属性维度.
const int ClusterK=8;//聚成的簇的数量
typedef  double AttributeType;
struct Object{//数据项的数据结构
    AttributeType attribute[AttributeCount];
};
vector<Object> allObj;//保存所有的数据
Object cluster[1000][ClusterK];//各个簇的数据项，这里假定每个簇的最大量为1000了，可以写成vector的数据结构，
Object oldcenter[ClusterK];//旧的各个中心点
int oldCluObjNum[ClusterK];//旧的各个簇有多少数据量
Object center[ClusterK];//对比旧的中心点
int CluObjNum[ClusterK];//对比旧的各簇的数据量
void getAllobject(ifstream &ifs);//加载所有数据
void kmeans(ifstream &ins);//算法
void produceData(string fileName,int maxNum,int objectNum);//随机产生数据，fileName文件名，maxNum数据的最大数，objectNum数据个数
int cloestCluster(Object obj);//返回当前数据项与哪个簇最近
void initCenter();//初始化各中心点
void updateCluster(int cluK,Object obj);//更新簇结构
bool isChange();//判断迭代之后中心点是否改变，若没有改变可以迭代结束了，得到局部最优解
void copyCenter();//复制到旧的中
void computeCenter();//重新计算中心点
AttributeType Distance(Object obj,Object obj2);//计算两个点之间的距离
int main(){
    //produceData("data2.txt",100,50);
    ifstream ifs;
    ifs.open("data2.txt");
    kmeans(ifs);
    ifs.close();
    system("pause");
}

void kmeans(ifstream &ins){
    getAllobject(ins);
    initCenter();
    for(int i=0;i<ClusterK;i++){
        center[i]=allObj[i];
        CluObjNum[i]=0;
    }
    int repeat=0;
    while(isChange()&&repeat<repeatMax){//一直迭代，直到中心点不再改变，或者达到迭代的上限
        copyCenter();
        for(vector<Object>::iterator begin=allObj.begin();begin<allObj.end();begin++){
            int closestK=cloestCluster(*begin);
            updateCluster(closestK,*begin);
        }
        computeCenter();
        for(int i=0;i<ClusterK;i++){
            cout<<"第"<<i<<"个簇,他们之间的中心点是:";
            char file[]={‘c‘,‘l‘,‘u‘,‘s‘,‘t‘,‘e‘,‘r‘,static_cast<char>(i+‘0‘),‘.‘,‘t‘,‘x‘,‘t‘,‘\0‘};
            ofstream out;
            out.open(file,ifstream::trunc);//输入到各个簇的文件中保存
            for(int l=0;l<AttributeCount;l++){
                cout<<center[i].attribute[l]<<" ";
            }
            cout<<endl;
            for(int m=1;m<=CluObjNum[i];m++){
                for(int j=0;j<AttributeCount;j++)
                    out<<cluster[m][i].attribute[j]<<" ";
                out<<endl;
            }
            cout<<endl;
            out.close();
        }
        cout<<endl;
        repeat++;
    }
}
void updateCluster(int cluK,Object obj){//把obj更新到cluK簇中，同时项增加1
    cluster[CluObjNum[cluK]+1][cluK]=obj;
    CluObjNum[cluK]++;
}
void computeCenter(){
    for(int i=0;i<ClusterK;i++){
        for(int m=0;m<AttributeCount;m++){
            double sum=0;
            for(int j=0;j<CluObjNum[i];j++){
                sum+=cluster[j][i].attribute[m];
            }
            center[i].attribute[m]=sum/CluObjNum[i];
        }
    }
}
void copyCenter(){
    for(int i=0;i<ClusterK;i++){
        oldCluObjNum[i]=CluObjNum[i];
        CluObjNum[i]=0;
        for(int j=0;j<AttributeCount;j++){
            oldcenter[i].attribute[j]=center[i].attribute[j];
        }
    }
}
void initCenter(){
    Object obj;
    for(int i=0;i<AttributeCount;i++){
        obj.attribute[i]=-1;
    }
    for(int i=0;i<ClusterK;i++){
            oldcenter[i]=obj;
    }
}
int cloestCluster(Object obj){
    AttributeType sq=maxNum,m=maxNum;
    int theCloest=0;
    for(int i=0;i<ClusterK;i++){
        m=Distance(obj,center[i]);
        if(m<sq){
            theCloest=i;
            sq=m;
        }
    }
    return theCloest;
}
AttributeType Distance(Object obj,Object obj2){
    AttributeType dis=0;
    for(int i=0;i<AttributeCount;i++){
        dis+=(obj.attribute[i]-obj2.attribute[i])*(obj.attribute[i]-obj2.attribute[i]);
    }
    return dis;
}

bool isChange(){
    for(int i=0;i<ClusterK;i++){
        for(int j=0;j<AttributeCount;j++)
            if(oldcenter[i].attribute[j]!=center[i].attribute[j])
                return true;
    }
    return false;
}
void getAllobject(ifstream &ifs){
    while(ifs){
        Object obj;
        for(int i=0;i<AttributeCount;i++)
            ifs>>obj.attribute[i];
        allObj.push_back(obj);
    }
}