Weka K-means距离

时间:2015-03-29 19:19:07

标签: java weka distance k-means

我使用weka库来使用SimpleKMeans函数。

我的arff文件是:

@relation digits

@attribute number numeric

@data
3.708699941635132
3.608700037002563
3.508699893951416
3.808700084686279
3.708699941635132
3.708699941635132
3.708699941635132
3.708699941635132
3.708699941635132
3.408699989318847
3.708699941635132

它是质心,我也有距离矩阵,包含所有质心距离(有特殊的计算距离,而不是简单的欧几里德距离)。如何在代码中传输距离矩阵?现在我训练这段代码:

package kmeanstest;

import java.io.BufferedReader;
import java.io.FileReader;
import weka.clusterers.SimpleKMeans;
import weka.core.Instances;

public class Kmeanstest {
    public Kmeanstest() throws Exception {
        BufferedReader breader = new BufferedReader(new FileReader("data.arff"));
        Instances Train = new Instances(breader);
        SimpleKMeans kMeans = new SimpleKMeans();
        kMeans.setSeed(10);
        kMeans.setPreserveInstancesOrder(true);
        kMeans.setNumClusters(3);
        kMeans.buildClusterer(Train);
        int[] assignments = kMeans.getAssignments();
        int i = 0;
        for (int clusterNum : assignments) {
            System.out.println("Instance " + i + " -> Cluster " + clusterNum);
            i++;
        }
        breader.close();
    }
    public static void main(String[] args) throws Exception {
        Kmeanstest kmeanstest = new Kmeanstest();
    }
}

1 个答案:

答案 0 :(得分:1)

在我的项目中,我采用了类似的结果来比较这些距离函数。(在我的例子中,我有40.000个实例和10个特征)。

但是,如果您使用2个以上的功能,最好创建自己的距离函数(例如,汉密尔顿距离。我坚信它会有更好的结果)。

ManhattanDistance manhattan = new ManhattanDistance();
    try {
        kmeans.setDistanceFunction(manhattan);
    } catch (Exception e2) {
        // TODO Auto-generated catch block
        e2.printStackTrace();
    }