cc.mallet.pipe.Target2Integer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mallet 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.

There is a newer version: 2.0.12

Show newest version

/* Copyright (C) 2002 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.pipe;

import cc.mallet.types.Instance;

/** Convert object in the target field into an integer numeric type
 *   @author David Mimno
 */

public class Target2Integer extends Pipe {

	public Instance pipe (Instance carrier) {
		if (carrier.getTarget() != null) {
			if (! (carrier.getTarget() instanceof String)) {
				throw new IllegalArgumentException ("Target must be a String for conversion to Integer");
			}
			carrier.setTarget( new Integer((String) carrier.getTarget()) );
		}
		return carrier;
	}

}