read.java

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

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

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


public class Read {
    static public void main(String[] argv) 
	throws IOException, FileFormatException {
	
	System.out.println("----------------- Reading multiple sequences ---");
	System.out.println("---- Integer ----");
	testInteger();
	
	System.out.println("---- Vector ----");
	testVector();

	System.out.println("----------------- Reading one sequence ---------");
	testOne();
    }

    static void testInteger() 
	throws IOException, FileFormatException {
	Reader reader = new FileReader("testInteger.seq");
	Vector v = ObservationSequencesReader.
	    readSequences(new ObservationIntegerReader(), reader);
	reader.close();
	
	System.out.println("Number of sequences read: " + v.size());
	System.out.println("First observation: " + 
			   ((Vector) v.elementAt(0)).elementAt(0));
    }

    static void testOne() 
	throws IOException, FileFormatException {
	Reader reader = new FileReader("testInteger.seq");
	Vector v = ObservationSequencesReader.
	    readSequence(new ObservationIntegerReader(), reader);
	reader.close();
	
	System.out.println("Sequence length: " + v.size());
    }

    static void testVector() 
	throws IOException, FileFormatException {
	Reader reader = new FileReader("testVector.seq");
	Vector v = ObservationSequencesReader.
	    readSequences(new ObservationVectorReader(2), reader);
	reader.close();
	
	System.out.println("Number of sequences read: " + v.size());
	System.out.println("First observation: " + 
			   ((Vector) v.elementAt(0)).elementAt(0));

	KMeansLearner kml = 
	    new KMeansLearner(3, new OpdfMultiGaussianFactory(2), v);
	Hmm fittedHmm = kml.learn();

	HmmDrawer hmmDrawer = new HmmDrawer();
	hmmDrawer.write(fittedHmm, "hmm_io.dot");
    }
}