sequenceprintingpipe.java

来自「mallet是自然语言处理、机器学习领域的一个开源项目。」· Java 代码 · 共 75 行

/* Copyright (C) 2003 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 edu.umass.cs.mallet.base.pipe.tsf;

import edu.umass.cs.mallet.base.pipe.Pipe;
import edu.umass.cs.mallet.base.types.*;
import edu.umass.cs.mallet.base.util.PropertyList;
import edu.umass.cs.mallet.base.util.Maths;

import java.io.PrintWriter;

/**
 * Created: Jul 6, 2005
 *
 * @author <A HREF="mailto:[e-mail]>[e-mail]</A>
 * @version $Id: SequencePrintingPipe.java,v 1.2 2005/07/08 20:37:26 casutton Exp $
 */
public class SequencePrintingPipe extends Pipe {

  private PrintWriter writer;

  public SequencePrintingPipe (PrintWriter writer)
  {
    this.writer = writer;
  }

  public Instance pipe (Instance carrier)
  {
    Sequence data = (Sequence) carrier.getData ();
    Sequence target = (Sequence) carrier.getTarget ();

    if (data.size () != target.size ())
      throw new IllegalArgumentException ("Trying to print into SimpleTagger format, where data and target lengths do not match\n"
      +"data.length = "+data.size()+", target.length = "+target.size ());

    int N = data.size ();

    if (data instanceof TokenSequence) {
      throw new UnsupportedOperationException ("Not yet implemented.");
    } else if (data instanceof FeatureVectorSequence) {

      FeatureVectorSequence fvs = (FeatureVectorSequence) data;
      Alphabet dict = (fvs.size() > 0) ? fvs.getFeatureVector (0).getAlphabet () : null;

      for (int i = 0; i < N; i++) {
        Object label = target.get (i);
        writer.print (label);

        FeatureVector fv = fvs.getFeatureVector (i);
        for (int loc = 0; loc < fv.numLocations (); loc++) {
          writer.print (' ');
          String fname = dict.lookupObject (fv.indexAtLocation (loc)).toString ();
          double value = fv.valueAtLocation (loc);
          if (!Maths.almostEquals (value, 1.0)) {
            throw new IllegalArgumentException ("Printing to SimpleTagger format: FeatureVector not binary at time slice "+i+" fv:"+fv);
          }
          writer.print (fname);
        }
        writer.println ();
      }
    } else {
      throw new IllegalArgumentException ("Don't know how to print data of type "+data);
    }

    writer.println ();

    return carrier;
  }

}