kmeans.java

来自「java实现的隐马尔科夫模型」· Java 代码 · 共 50 行

package be.ac.ulg.montefiore.run.jahmm.test;

import be.ac.ulg.montefiore.run.jahmm.*;
import be.ac.ulg.montefiore.run.jahmm.learn.*;
import java.util.*;


public class KMeans {
    static final int nbStates = 5;
    
    static public void main(String[] argv) {
	
	Vector oseq1 = new Vector();
	for (int i = 0; i < 5; i++)
	    oseq1.add(new ObservationInteger(i));

	Vector oseq2 = new Vector();
	for (int i = 4; i < 10; i++)
	    oseq2.add(new ObservationInteger(i));

	Vector obsSeqs = new Vector();
	obsSeqs.add(oseq1);
	obsSeqs.add(oseq2);

	KMeansLearner kml = 
	    new KMeansLearner(nbStates, new OpdfIntegerFactory(10), obsSeqs);
	
	kml.iterate();
    }

    static private String toString(int[] sseq) {
	String s = "[ ";
	
	for (int i = 0; i < sseq.length; i++)
	    s += sseq[i] + " ";

	return s + "]";
    }

    static private String toString(Vector cluster) {
	String s = "[ ";
	
	for (int i = 0; i < cluster.size(); i++) {
	    s += cluster.elementAt(i).toString() + " ";
	}

	return s + "]";
    }
}