Question

我想测量一下在OpenIMAJ库中KMColourSegmenter执行聚类所花费的时间。

如果我没有使初始质心固定，而不是随机，我无法使测量表现出来;因为它每次都会改变，给出不同的迭代次数，并且可以随时间变化来执行聚类。

那么如何修复初始质心，即手动设置它们？

更新

@Jon感谢您的回答，我正在尝试实施您所说的内容。你能检查一下吗，特别是“clusters”数组我认为这个数组初始化没有意义。如果我错了，请纠正我。

public class MyFloatKMeansInit extends FloatKMeansInit{

    @Override
    public void initKMeans(DataSource<float[]> bds, float[][] clusters) throws IOException {
        // TODO Auto-generated method stub
        for (int i = 0; i < bds.size(); i++) {
            for (int j = 0; j < bds.getData(i).length; j++) {
                clusters[i][j]=bds.getData(i)[j];
            }
        }   
    }
}

public class MyKMColourSegmenter extends KMColourSegmenter{

    public MyKMColourSegmenter(FloatArrayBackedDataSource bds, ColourSpace colourSpace, int K) throws IOException {
        super(colourSpace, K);
        MyFloatKMeansInit myFloatKMeansInit = new MyFloatKMeansInit();
        float[][] clusters = new float[K][];//#######I think there is something wrong here
        myFloatKMeansInit.initKMeans(bds, clusters);
        this.kmeans.setInit(myFloatKMeansInit);
        // TODO Auto-generated constructor stub
    }

}

Answer 1

你必须自己实施;创建一个KMColourSegmenter的子类，并创建一个接受质心的构造函数，以及您在KMColourSegmenter超类中选择构造函数所需的任何参数。然后在构造函数中，在调用super之后，使用this.kmeans.setInit()方法设置初始化以使用预定义的质心。您需要实现一个自定义FloatKMeansInit子类，它允许您在外部设置质心，但这应该是微不足道的，因为它只需要实现一个方法。

针对修正后的问题进行更新：

您不应直接致电initKMeans;运行算法时在幕后发生的事情。您还需要使用所需的质心初始质心填充质心数组，而不是bds。例如（未经测试）：

public class MyFloatKMeansInit extends FloatKMeansInit{
    private float [][] mycentroids;

    public MyFloatKMeansInit(float [][] mycentroids) {//modifying data type here
        this.mycentroids = mycentroids;
    }

    @Override
    public void initKMeans(DataSource<float[]> bds, float[][] clusters) throws IOException {
        for (int i = 0; i < mycentroids.length; i++) {
            for (int j = 0; j < mycentroids[i].length; j++) {
                clusters[i][j]=mycentroids[i][j]; //could use arraycopy instead
            }
        }   
    }
}

public class MyKMColourSegmenter extends KMColourSegmenter{

    public MyKMColourSegmenter(ColourSpace colourSpace, float[][] mycentroids) throws IOException {
        super(colourSpace, mycentroids.length);
        MyFloatKMeansInit myFloatKMeansInit = new MyFloatKMeansInit(mycentroids);
        this.kmeans.setInit(myFloatKMeansInit);
    }

}

如何在OpenIMAJ库中设置KMColourSegmenter的初始质心？

1 个答案: