weka使用笔记1-cluster内方法的使用

weka提供了11种聚类算法，常用的有kmeans 和EM，weka的simpleKmeans是最简单的聚类方法，距离是计算欧几里德距离，但是新版本的weka是可以置入自己的计算距离的公式的。EM聚类的效率很低，建议如果不是实验情况，不要使用。

主要说一下simpleKmeans聚类。SimpleKmeans聚类参数很简单，这里不再赘述，文档里面写的很清楚。用图形界面的情况下，聚类之后，在左下方的result list栏里，右键，有很多选项，其中visualize cluster assignment可以看聚类之后的结果，在这个visualize界面，有一个save，是可以把类标号添加到每一行数据的，如果数据里面有用户ID的话，那这个就可以把每个用户ID和这个用户所在的类提取出来，很方便，生成的也是.arff文件。

但是在命令行模式下，没有办法得到这个类标号，只能得到一个很简单的数据，所以只能用编程实现了。

在一个工程内引用weka.jar之后就能用weka的接口了，非常方便.废话不说，直接上代码：

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileWriter;

import java.io.IOException;



import weka.clusterers.SimpleKMeans;

import weka.clusterers.ClusterEvaluation;

import weka.core.Instances;

import weka.core.converters.ArffLoader;



public class Kmeans {



	public static void startCluster(String inputfilename,String outputfilename) {

		Instances ins = null;

		SimpleKMeans KM = null;

		ClusterEvaluation cl = null;

		File file = new File(inputfilename);

		ArffLoader loader = new ArffLoader();

		File f = new File(outputfilename);

		try {

			cl = new ClusterEvaluation();	

			loader.setFile(file);

			ins = loader.getDataSet();

			ins.deleteAttributeAt(0);//忽略第一列

			KM = new SimpleKMeans();



			KM.setNumClusters(15);

			KM.buildClusterer(ins);	

			cl.setClusterer(KM);

			cl.evaluateClusterer(new Instances(ins));

			double[] d = cl.getClusterAssignments();//得到每一列的类标号

			

			if (f.exists()) {

				System.out.print("cunzai");

			} else {

				System.out.print("bucunzai");

				f.createNewFile();

			}

			BufferedWriter output = new BufferedWriter(new FileWriter(f));

			String dataline = "";

			output.write(cl.clusterResultsToString());//就是聚类之后打印出来的聚类结果

			for(int i = 0 ; i < d.length;i++){

				dataline = d[i]+"";

				output.write(dataline);

				output.newLine();

				System.out.println(dataline);

			}

			

			output.close();



		} catch (IOException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		} catch (Exception e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

	}

}

　　然后打上jar包就能运行了，上面代码只是得到了一个简单的类标号，还可以根据自己的需要输出更多的数据。