SequenceLoss.java

package edu.fudan.ml.loss.struct;

import edu.fudan.ml.loss.Loss;

public class SequenceLoss implements Loss {
	/**
	 * 
	 * @author xpqiu
	 *
	 */
	public static enum Type	{
		POINT, EDGE
	}
	
	Type type;
	
	public SequenceLoss(Type type)	{
		this.type = type;
	}

	public float calc(Object o1, Object o2) {
		
		float errCount = 0;
		if (o1 instanceof int[] && o2 instanceof int[]) {
			int[] pred = (int[]) o1;
			int[] gold = (int[]) o2;

			if (type == Type.POINT)	{
				for (int i = 0; i < pred.length; i++) {
					if (pred[i] != gold[i])
						errCount++;
				}
			}else if (type == Type.EDGE)	{
				for (int i = 1; i < pred.length; i++) {
					if (pred[i - 1] != gold[i - 1] || pred[i] != gold[i])
						errCount++;
				}
			}
		}

		return errCount;
	}

}