cc.mallet.grmm.util.ModelWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jcore-mallet-2.0.9 Show documentation

MALLET is a Java-based package for statistical natural language processing, document classification, clustering, topic modeling, information extraction, and other machine learning applications to text.

The newest version!

/* Copyright (C) 2006 Univ. of Massachusetts Amherst, Computer Science Dept.
   This file is part of "MALLET" (MAchine Learning for LanguagE Toolkit).
   http://www.cs.umass.edu/~mccallum/mallet
   This software is provided under the terms of the Common Public License,
   version 1.0, as published by http://www.opensource.org.  For further
   information, see the file `LICENSE' included with this distribution. */
package cc.mallet.grmm.util;


import java.io.*;
import cc.mallet.grmm.types.*;

/**
 * $Id: ModelReader.java,v 1.1 2007/10/22 21:37:58 mccallum Exp $
 */
public class ModelWriter {

    public static void writeModel (FactorGraph fg, Writer w) 
    {
	try {
	    writeVariables (fg, w);
	    w.write ("\n");
	    writeFactors (fg, w);
	} catch (Exception e) {
	    throw new RuntimeException (e);
	}
    }

    private static void writeVariables (FactorGraph fg, Writer w)  throws IOException
    {
	for (int vi = 0; vi < fg.numVariables(); vi++) {
	    Variable var = fg.getVariable (vi);
	    int nOuts = var.getNumOutcomes ();
	    String outStr = nOuts == Variable.CONTINUOUS ? "continuous" : Integer.toString(nOuts);
	    w.write("VAR " + var.getLabel() + " : " + outStr + "\n");
	}
    }

    private static void writeFactors (FactorGraph fg, Writer w) throws IOException
    {
	for (int fi = 0; fi < fg.factors().size(); fi++) {
	    Factor f = fg.getFactor (fi);
	    w.write (f.prettyOutputString ());
	    w.write ("\n");
	}
    }

}