cc.mallet.fst.confidence.InstanceWithConfidence 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) 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. */

/** 
		@author Aron Culotta [email protected]
*/

package cc.mallet.fst.confidence;

import cc.mallet.types.*;

/**
	 Helper class to store confidence of an Instance.
 */
public class InstanceWithConfidence implements Comparable{
	double confidence;
	Instance instance;
	boolean correct;
	
	public InstanceWithConfidence (Instance inst, double c, Sequence predicted) {
		this.instance = inst;
		this.confidence = c;
		this.correct = true;
		Sequence truth = (Sequence) inst.getTarget ();
		for (int i=0; i < truth.size(); i++) {
			if (!truth.get(i).equals (predicted.get(i))) {
				this.correct = false;
				break;
			}				
		}
	}

	public InstanceWithConfidence (Instance inst, double c, boolean correct) {
		this.instance = inst;
		this.confidence = c;
		this.correct = correct;
	}

	public int compareTo (Object o) {
		InstanceWithConfidence inst = (InstanceWithConfidence) o;
		if (this.confidence > inst.confidence)
			return 1;
		else if (this.confidence < inst.confidence)
			return -1;
		else return 0;
	}

	public double getConfidence () { return confidence; }
	public Instance getInstance () { return instance; }
	public boolean correct () { return correct; }
}