K-means (PRML) in C++
#include <iostream>
#include <fstream>
#include <sstream>
#include <vector>
#include <string>
#include <algorithm>
#include <numeric>
#include <cmath>
#include <limits>
template <class T>
void ReadDataFromFile(const std::string &filename, std::vector<std::vector<T> > &vv_data) {
std::ifstream vm_info(filename.c_str());
T x, y;
std::vector<T> v_data;
while(!vm_info.eof()) {
v_data.clear();
vm_info >> x >> y;
v_data.push_back(x);
v_data.push_back(y);
vv_data.push_back(v_data);
}
vm_info.close();
}
template <class T>
void Display2DVector(std::vector<std::vector<T> > &vv) {
for(size_t i=0;i<vv.size();++i) {
for(typename::std::vector<T>::const_iterator it=vv.at(i).begin();it!=vv.at(i).end();++it) {
std::cout<<*it<<" ";
}
std::cout<<"\n";
}
std::cout<<"--------the total of the 2DVector is "<<vv.size()<<std::endl;
}
template <class T>
void AddIndicator(std::vector<std::vector<T> > &vv, const int &k) {
for(size_t i=0; i<vv.size(); ++i) {
for(size_t j=0; j<k; ++j) {
vv.at(i).push_back(0);
}
}
}
template <class T1, class T2>
void UpdateIndicator(std::vector<std::vector<T1> > &vv, const std::vector<T2> &u, const int &k) {
for(size_t i=0; i<vv.size(); ++i) {
double dis=std::numeric_limits<double>::max(), dis_min, cluster;
for(size_t j=0; j<k; ++j) {
dis_min=pow(vv.at(i).at(0)-u.at(j*2), 2.0)+pow(vv.at(i).at(1)-u.at(j*2+1), 2.0);
if(dis_min < dis) {
dis=dis_min;
cluster=j;
}
}
vv.at(i).at(cluster+2)=1;
}
}
template <class T1, class T2>
void UpdateMeans(const std::vector<std::vector<T1> > &vv, std::vector<T2> &u, const int &k) {
std::vector<T2> sum_set(u.size(), 0);
for(size_t i=0; i<k; ++i) {
double sum_indi=0.0;
for(size_t j=0; j<vv.size(); ++j) {
sum_indi+=vv.at(j).at(i+2);
sum_set.at(i*2)+=vv.at(j).at(i+2)*vv.at(j).at(0);
sum_set.at(i*2+1)+=vv.at(j).at(i+2)*vv.at(j).at(1);
}
sum_set.at(i*2)/=sum_indi;
sum_set.at(i*2+1)/=sum_indi;
}
u=sum_set;
}
template <class T1, class T2>
double DistortionMeasure(const std::vector<std::vector<T1> > &vv, const std::vector<T2> &u, const int &k) {
double cost=0.0;
for(size_t i=0; i<vv.size(); ++i) {
for(size_t j=0; j<k; ++j) {
cost+=vv.at(i).at(j+2)*(pow(vv.at(i).at(0)-u.at(j*2), 2.0)+pow(vv.at(i).at(1)-u.at(j*2+1), 2.0));
}
}
return cost;
}
int main() {
int k=4;
double mean[]={39, 42, 70, 2, 230, 10, 190, 85};
std::vector<double> u(mean, mean+sizeof(mean)/sizeof(mean[0]));
std::string oridata="kmeans.dat";
std::vector<std::vector<double> > vv_data;
ReadDataFromFile(oridata, vv_data);
AddIndicator(vv_data, k);
std::cout<<"the original mean: \n";
for(std::vector<double>::const_iterator it=u.begin(); it!=u.end(); ++it) {
std::cout<<*it<<" ";
}
std::cout<<std::endl;
double cost_old=std::numeric_limits<double>::max();
while(true) {
double cost_new=DistortionMeasure(vv_data, u, k);
if(std::abs(cost_new-cost_old)<0.0000001)
break;
UpdateIndicator(vv_data, u, k);
UpdateMeans(vv_data, u, k);
cost_old=cost_new;
}
std::cout<<"the new mean: \n";
for(std::vector<double>::const_iterator it=u.begin(); it!=u.end(); ++it) {
std::cout<<*it<<" ";
}
std::cout<<std::endl;
return 0;
}
The two phases of re-assigning data points to clusters and re-computing the cluster means are repeated in turn until there is no further change in the assignments(or until some maximum number of iterations is exceeded).
K-means (PRML) in C++的更多相关文章
- KNN 与 K - Means 算法比较
KNN K-Means 1.分类算法 聚类算法 2.监督学习 非监督学习 3.数据类型:喂给它的数据集是带label的数据,已经是完全正确的数据 喂给它的数据集是无label的数据,是杂乱无章的,经过 ...
- 软件——机器学习与Python,聚类,K——means
K-means是一种聚类算法: 这里运用k-means进行31个城市的分类 城市的数据保存在city.txt文件中,内容如下: BJ,2959.19,730.79,749.41,513.34,467. ...
- 快速查找无序数组中的第K大数?
1.题目分析: 查找无序数组中的第K大数,直观感觉便是先排好序再找到下标为K-1的元素,时间复杂度O(NlgN).在此,我们想探索是否存在时间复杂度 < O(NlgN),而且近似等于O(N)的高 ...
- 网络费用流-最小k路径覆盖
多校联赛第一场(hdu4862) Jump Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Ot ...
- numpy.ones_like(a, dtype=None, order='K', subok=True)返回和原矩阵一样形状的1矩阵
Return an array of ones with the same shape and type as a given array. Parameters: a : array_like Th ...
- 当我们在谈论kmeans(2)
本稿为初稿,后续可能还会修改:如果转载,请务必保留源地址,非常感谢! 博客园:http://www.cnblogs.com/data-miner/ 其他:建设中- 当我们在谈论kmeans(2 ...
- scikit-learn包的学习资料
http://scikit-learn.org/stable/modules/clustering.html#k-means http://my.oschina.net/u/175377/blog/8 ...
- HDU 3584 Cube (三维 树状数组)
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=3584 Cube Problem Description Given an N*N*N cube A, ...
- Torch7学习笔记(二)nn Package
神经网络Package [目前还属于草稿版,等我整个学习玩以后会重新整理] 模块Module module定义了训练神经网络需要的所有基础方法,并且是可以序列化的抽象类. module有两种状态变量: ...
- 2016中国大学生程序设计竞赛 - 网络选拔赛 J. Alice and Bob
Alice and Bob Time Limit: 10000/5000 MS (Java/Others) Memory Limit: 131072/131072 K (Java/Others) ...
随机推荐
- CSS——滑动门
在背景图片中可以对图片进行圆角设置,但是这样是写死的.如下图: 情况分析:如果我们li标签中的文字变少了或者变多了,我们就需要重新定义背景图片.所以我们使用滑动门技术.它将图片特殊地方进行分割.宽度利 ...
- javascript之console篇
javascript中的console使用得当,将会事半功倍,对bug,性能等的跟踪,优化是个不错的利器! 1.基本日志消息打印: console.debug(msg); console.info() ...
- Caffe2:添加CUDA路径
在使用Eclipse时候,仍然出现 libcuda.so 找不到的情况..... 原因:cuda环境没有设置到系统环境中. 方案: 设置CUDA path: 在bashrc中添加路径,只对当前用户产生 ...
- CSS3利用box-shadow实现相框效果
CSS3利用box-shadow实现相框效果 <style> html { overflow: hidden; background-color: #653845; background- ...
- Android控件的继承关系
1.View,ViewGroup >View: }1.所有高级UI组件都继承View类而实现的 }2.一个View在屏幕上占据一块矩形区域 }3. 负责渲染 }4.负责处理发生的事件 }5.设置 ...
- PHP 之base16加密解密封装
/** * @Description: base16加密 * @Author: Yang * @param $data * @return string */ function base16_enco ...
- S3C2440时钟体系
注:以下内容学习于韦东山老师arm裸机第一期视频教程 一. 2440时钟简介 1.1 2440是一个SOC(system on chip)系统,不仅有很多CPU,还有很多外设,在2440芯片手册有系统 ...
- IDEA生成增强for循环
itar 生成array for代码块 for (int i = 0; i < array.length; i++) { = array[i]; } itco 生成Collection迭代 fo ...
- Linux - redis主从同步
目录 Linux - redis主从同步 环境准备 配置主从同步 测试写入数据,主库写入数据,检查从库数据 手动进行主从复制故障切换 Linux - redis主从同步 原理: 从服务器向主服务器发送 ...
- 【codeforces 701C】They Are Everywhere
[题目链接]:http://codeforces.com/contest/701/problem/C [题意] 让你选择一段最短的区间; 使得这段区间里面包含所有种类的字符; [题解] 之前都是用二分 ...