K均值聚類(lèi)算法的Java版實(shí)現(xiàn)代碼示例
更新時(shí)間：2017年12月08日 16:48:04 作者：cyxlzzs
這篇文章主要介紹了K均值聚類(lèi)算法的Java版實(shí)現(xiàn)代碼示例，具有一定借鑒價(jià)值，需要的朋友可以參考下。
1.簡(jiǎn)介
K均值聚類(lèi)算法是先隨機(jī)選取K個(gè)對(duì)象作為初始的聚類(lèi)中心。然后計(jì)算每個(gè)對(duì)象與各個(gè)種子聚類(lèi)中心之間的距離，把每個(gè)對(duì)象分配給距離它最近的聚類(lèi)中心。聚類(lèi)中心以及分配給它們的對(duì)象就代表一個(gè)聚類(lèi)。一旦全部對(duì)象都被分配了，每個(gè)聚類(lèi)的聚類(lèi)中心會(huì)根據(jù)聚類(lèi)中現(xiàn)有的對(duì)象被重新計(jì)算。這個(gè)過(guò)程將不斷重復(fù)直到滿(mǎn)足某個(gè)終止條件。終止條件可以是沒(méi)有（或最小數(shù)目）對(duì)象被重新分配給不同的聚類(lèi)，沒(méi)有（或最小數(shù)目）聚類(lèi)中心再發(fā)生變化，誤差平方和局部最小。
2.什么是聚類(lèi)
聚類(lèi)是一個(gè)將數(shù)據(jù)集中在某些方面相似的數(shù)據(jù)成員進(jìn)行分類(lèi)組織的過(guò)程，聚類(lèi)就是一種發(fā)現(xiàn)這種內(nèi)在結(jié)構(gòu)的技術(shù)，聚類(lèi)技術(shù)經(jīng)常被稱(chēng)為無(wú)監(jiān)督學(xué)習(xí)。
3.什么是k均值聚類(lèi)
k均值聚類(lèi)是最著名的劃分聚類(lèi)算法，由于簡(jiǎn)潔和效率使得他成為所有聚類(lèi)算法中最廣泛使用的。給定一個(gè)數(shù)據(jù)點(diǎn)集合和需要的聚類(lèi)數(shù)目k，k由用戶(hù)指定，k均值算法根據(jù)某個(gè)距離函數(shù)反復(fù)把數(shù)據(jù)分入k個(gè)聚類(lèi)中。
4.實(shí)現(xiàn)
Java代碼如下：
package org.algorithm;
import java.util.ArrayList;
import java.util.Random;
/**
* K均值聚類(lèi)算法
*/
public class Kmeans {
private int k;
// 分成多少簇
private int m;
// 迭代次數(shù)
private int dataSetLength;
// 數(shù)據(jù)集元素個(gè)數(shù)，即數(shù)據(jù)集的長(zhǎng)度
private ArrayList<float[]> dataSet;
// 數(shù)據(jù)集鏈表
private ArrayList<float[]> center;
// 中心鏈表
private ArrayList<ArrayList<float[]>> cluster;
// 簇
private ArrayList<float> jc;
// 誤差平方和，k越接近dataSetLength，誤差越小
private Random random;
/**
* 設(shè)置需分組的原始數(shù)據(jù)集
*
* @param dataSet
*/
public void setDataSet(ArrayList<float[]> dataSet) {
this.dataSet = dataSet;
}
/**
* 獲取結(jié)果分組
*
* @return 結(jié)果集
*/
public ArrayList<ArrayList<float[]>> getCluster() {
return cluster;
}
/**
* 構(gòu)造函數(shù)，傳入需要分成的簇?cái)?shù)量
*
* @param k
* 簇?cái)?shù)量,若k<=0時(shí)，設(shè)置為1，若k大于數(shù)據(jù)源的長(zhǎng)度時(shí)，置為數(shù)據(jù)源的長(zhǎng)度
*/
public Kmeans(int k) {
if (k <= 0) {
k = 1;
}
this.k = k;
}
/**
* 初始化
*/
private void init() {
m = 0;
random = new Random();
if (dataSet == null || dataSet.size() == 0) {
initDataSet();
}
dataSetLength = dataSet.size();
if (k > dataSetLength) {
k = dataSetLength;
}
center = initCenters();
cluster = initCluster();
jc = new ArrayList<float>();
}
/**
* 如果調(diào)用者未初始化數(shù)據(jù)集，則采用內(nèi)部測(cè)試數(shù)據(jù)集
*/
private void initDataSet() {
dataSet = new ArrayList<float[]>();
// 其中{6,3}是一樣的，所以長(zhǎng)度為15的數(shù)據(jù)集分成14簇和15簇的誤差都為0
float[][] dataSetArray = new float[][] { { 8, 2 }, { 3, 4 }, { 2, 5 },
{ 4, 2 }, { 7, 3 }, { 6, 2 }, { 4, 7 }, { 6, 3 }, { 5, 3 },
{ 6, 3 }, { 6, 9 }, { 1, 6 }, { 3, 9 }, { 4, 1 }, { 8, 6 } };
for (int i = 0; i < dataSetArray.length; i++) {
dataSet.add(dataSetArray[i]);
}
}
/**
* 初始化中心數(shù)據(jù)鏈表，分成多少簇就有多少個(gè)中心點(diǎn)
*
* @return 中心點(diǎn)集
*/
private ArrayList<float[]> initCenters() {
ArrayList<float[]> center = new ArrayList<float[]>();
int[] randoms = new int[k];
Boolean flag;
int temp = random.nextint(dataSetLength);
randoms[0] = temp;
for (int i = 1; i < k; i++) {
flag = true;
while (flag) {
temp = random.nextint(dataSetLength);
int j = 0;
// 不清楚for循環(huán)導(dǎo)致j無(wú)法加1
// for(j=0;j<i;++j)
// {
// if(temp==randoms[j]);
// {
// break;
// }
// }
while (j < i) {
if (temp == randoms[j]) {
break;
}
j++;
}
if (j == i) {
flag = false;
}
}
randoms[i] = temp;
}
// 測(cè)試隨機(jī)數(shù)生成情況
// for(int i=0;i<k;i++)
// {
// System.out.println("test1:randoms["+i+"]="+randoms[i]);
// }
// System.out.println();
for (int i = 0; i < k; i++) {
center.add(dataSet.get(randoms[i]));
// 生成初始化中心鏈表
}
return center;
}
/**
* 初始化簇集合
*
* @return 一個(gè)分為k簇的空數(shù)據(jù)的簇集合
*/
private ArrayList<ArrayList<float[]>> initCluster() {
ArrayList<ArrayList<float[]>> cluster = new ArrayList<ArrayList<float[]>>();
for (int i = 0; i < k; i++) {
cluster.add(new ArrayList<float[]>());
}
return cluster;
}
/**
* 計(jì)算兩個(gè)點(diǎn)之間的距離
*
* @param element
* 點(diǎn)1
* @param center
* 點(diǎn)2
* @return 距離
*/
private float distance(float[] element, float[] center) {
float distance = 0.0f;
float x = element[0] - center[0];
float y = element[1] - center[1];
float z = x * x + y * y;
distance = (float) Math.sqrt(z);
return distance;
}
/**
* 獲取距離集合中最小距離的位置
*
* @param distance
* 距離數(shù)組
* @return 最小距離在距離數(shù)組中的位置
*/
private int minDistance(float[] distance) {
float minDistance = distance[0];
int minLocation = 0;
for (int i = 1; i < distance.length; i++) {
if (distance[i] < minDistance) {
minDistance = distance[i];
minLocation = i;
} else if (distance[i] == minDistance) // 如果相等，隨機(jī)返回一個(gè)位置
{
if (random.nextint(10) < 5) {
minLocation = i;
}
}
}
return minLocation;
}
/**
* 核心，將當(dāng)前元素放到最小距離中心相關(guān)的簇中
*/
private void clusterSet() {
float[] distance = new float[k];
for (int i = 0; i < dataSetLength; i++) {
for (int j = 0; j < k; j++) {
distance[j] = distance(dataSet.get(i), center.get(j));
// System.out.println("test2:"+"dataSet["+i+"],center["+j+"],distance="+distance[j]);
}
int minLocation = minDistance(distance);
// System.out.println("test3:"+"dataSet["+i+"],minLocation="+minLocation);
// System.out.println();
cluster.get(minLocation).add(dataSet.get(i));
// 核心，將當(dāng)前元素放到最小距離中心相關(guān)的簇中
}
}
/**
* 求兩點(diǎn)誤差平方的方法
*
* @param element
* 點(diǎn)1
* @param center
* 點(diǎn)2
* @return 誤差平方
*/
private float errorSquare(float[] element, float[] center) {
float x = element[0] - center[0];
float y = element[1] - center[1];
float errSquare = x * x + y * y;
return errSquare;
}
/**
* 計(jì)算誤差平方和準(zhǔn)則函數(shù)方法
*/
private void countRule() {
float jcF = 0;
for (int i = 0; i < cluster.size(); i++) {
for (int j = 0; j < cluster.get(i).size(); j++) {
jcF += errorSquare(cluster.get(i).get(j), center.get(i));
}
}
jc.add(jcF);
}
/**
* 設(shè)置新的簇中心方法
*/
private void setNewCenter() {
for (int i = 0; i < k; i++) {
int n = cluster.get(i).size();
if (n != 0) {
float[] newCenter = { 0, 0 };
for (int j = 0; j < n; j++) {
newCenter[0] += cluster.get(i).get(j)[0];
newCenter[1] += cluster.get(i).get(j)[1];
}
// 設(shè)置一個(gè)平均值
newCenter[0] = newCenter[0] / n;
newCenter[1] = newCenter[1] / n;
center.set(i, newCenter);
}
}
}
/**
* 打印數(shù)據(jù)，測(cè)試用
*
* @param dataArray
* 數(shù)據(jù)集
* @param dataArrayName
* 數(shù)據(jù)集名稱(chēng)
*/
public void printDataArray(ArrayList<float[]> dataArray,
String dataArrayName) {
for (int i = 0; i < dataArray.size(); i++) {
System.out.println("print:" + dataArrayName + "[" + i + "]={"
+ dataArray.get(i)[0] + "," + dataArray.get(i)[1] + "}");
}
System.out.println("===================================");
}
/**
* Kmeans算法核心過(guò)程方法
*/
private void kmeans() {
init();
// printDataArray(dataSet,"initDataSet");
// printDataArray(center,"initCenter");
// 循環(huán)分組，直到誤差不變?yōu)橹?
while (true) {
clusterSet();
// for(int i=0;i<cluster.size();i++)
// {
// printDataArray(cluster.get(i),"cluster["+i+"]");
// }
countRule();
// System.out.println("count:"+"jc["+m+"]="+jc.get(m));
// System.out.println();
// 誤差不變了，分組完成
if (m != 0) {
if (jc.get(m) - jc.get(m - 1) == 0) {
break;
}
}
setNewCenter();
// printDataArray(center,"newCenter");
m++;
cluster.clear();
cluster = initCluster();
}
// System.out.println("note:the times of repeat:m="+m);//輸出迭代次數(shù)
}
/**
* 執(zhí)行算法
*/
public void execute() {
long startTime = System.currentTimeMillis();
System.out.println("kmeans begins");
kmeans();
long endTime = System.currentTimeMillis();
System.out.println("kmeans running time=" + (endTime - startTime)
+ "ms");
System.out.println("kmeans ends");
System.out.println();
}
}
5.說(shuō)明：
具體代碼是從網(wǎng)上找的，根據(jù)自己的理解加了注釋和進(jìn)行部分修改，若注釋有誤還望指正
6.測(cè)試
package org.test;
import java.util.ArrayList;
import org.algorithm.Kmeans;
public class KmeansTest {
public static void main(String[] args)
{
//初始化一個(gè)Kmean對(duì)象，將k置為10
Kmeans k=new Kmeans(10);
ArrayList<float[]> dataSet=new ArrayList<float[]>();
dataSet.add(new float[]{1,2});
dataSet.add(new float[]{3,3});
dataSet.add(new float[]{3,4});
dataSet.add(new float[]{5,6});
dataSet.add(new float[]{8,9});
dataSet.add(new float[]{4,5});
dataSet.add(new float[]{6,4});
dataSet.add(new float[]{3,9});
dataSet.add(new float[]{5,9});
dataSet.add(new float[]{4,2});
dataSet.add(new float[]{1,9});
dataSet.add(new float[]{7,8});
//設(shè)置原始數(shù)據(jù)集
k.setDataSet(dataSet);
//執(zhí)行算法
k.execute();
//得到聚類(lèi)結(jié)果
ArrayList<ArrayList<float[]>> cluster=k.getCluster();
//查看結(jié)果
for (int i=0;i<cluster.size();i++)
{
k.printDataArray(cluster.get(i), "cluster["+i+"]");
}
}
}
總結(jié)：測(cè)試代碼已經(jīng)通過(guò)。并對(duì)聚類(lèi)的結(jié)果進(jìn)行了查看，結(jié)果基本上符合要求。至于有沒(méi)有更精確的算法有待發(fā)現(xiàn)。具體的實(shí)踐還有待挖掘
總結(jié)
以上就是本文關(guān)于K均值聚類(lèi)算法的Java版實(shí)現(xiàn)代碼示例的全部?jī)?nèi)容，希望對(duì)大家有所幫助。感興趣的朋友可以繼續(xù)參閱本站其他相關(guān)專(zhuān)題。如有不足之處，歡迎留言指出。感謝朋友們對(duì)本站的支持！
相關(guān)文章
java web學(xué)習(xí)_淺談request對(duì)象中g(shù)et和post的差異
下面小編就為大家?guī)?lái)一篇java web學(xué)習(xí)_淺談request對(duì)象中g(shù)et和post的差異。小編覺(jué)得挺不錯(cuò)的，現(xiàn)在就分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2017-05-05
springcloud gateway網(wǎng)關(guān)服務(wù)啟動(dòng)報(bào)錯(cuò)的解決
這篇文章主要介紹了springcloud gateway網(wǎng)關(guān)服務(wù)啟動(dòng)報(bào)錯(cuò)的解決，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-03-03
Jmeter參數(shù)化實(shí)現(xiàn)原理及過(guò)程解析
這篇文章主要介紹了Jmeter參數(shù)化實(shí)現(xiàn)原理及過(guò)程解析,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-07-07
java分頁(yè)之假分頁(yè)實(shí)現(xiàn)簡(jiǎn)單的分頁(yè)器
這篇文章主要介紹了java分頁(yè)之假分頁(yè)實(shí)現(xiàn)簡(jiǎn)單的分頁(yè)器的相關(guān)資料,需要的朋友可以參考下2016-04-04
Spring Security 實(shí)現(xiàn)“記住我”功能及原理解析
這篇文章主要介紹了Spring Security 實(shí)現(xiàn)“記住我”功能及原理解析,需要的朋友可以參考下2020-05-05
這篇文章主要介紹了spring springMVC中常用注解,本文給大家介紹的非常詳細(xì)，需要的朋友可以參考下2018-05-05
java注解是附加在代碼中的一些元信息，用于一些工具在編譯、運(yùn)行時(shí)進(jìn)行解析和使用，起到說(shuō)明、配置的功能,本文將詳細(xì)介紹，此功能的實(shí)現(xiàn)方法2012-11-11
已解決：No ''Access-Control-Allow-Origin''跨域問(wèn)題
這篇文章主要介紹了已解決：No 'Access-Control-Allow-Origin' 跨域，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-06-06
注解@TableName,@TableField,pgsql的模式對(duì)應(yīng)方式
這篇文章主要介紹了注解@TableName,@TableField,pgsql的模式對(duì)應(yīng)方式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-04-04
這篇文章主要給大家介紹了關(guān)于Java字段Stream排序常用方式的相關(guān)資料,我們?cè)谔幚頂?shù)據(jù)的時(shí)候經(jīng)常會(huì)需要進(jìn)行排序后再返回給前端調(diào)用,比如按照時(shí)間升序排序,前端展示數(shù)據(jù)就是按時(shí)間先后進(jìn)行排序,需要的朋友可以參考下2023-09-09