volumesmatchpipe.java

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

package edu.umass.cs.mallet.projects.seg_plus_coref.coreference;

import com.wcohen.secondstring.*;
import edu.umass.cs.mallet.base.types.*;
import edu.umass.cs.mallet.base.classify.*;
import edu.umass.cs.mallet.base.pipe.*;
import edu.umass.cs.mallet.base.pipe.iterator.*;
import edu.umass.cs.mallet.base.util.*;
import java.util.*;
import java.util.logging.*;
import java.lang.*;
import java.io.*;


/** feature is 1 if both string have things that look like
 * volumes/editions, but their numbers don't match*/
public class VolumesMatchPipe extends Pipe {

	private static Logger logger = MalletLogger.getLogger(VolumesMatchPipe.class.getName());

	
	public VolumesMatchPipe () {
	}


	public Instance pipe (Instance carrier) {
		NodePair pair = (NodePair)carrier.getData();
		Citation c1 = (Citation)pair.getObject1();
		Citation c2 = (Citation)pair.getObject2();

		String tok1 = getVenueOrEditionTokens (c1);
		String tok2 = getVenueOrEditionTokens (c2);
		
		if (tok1.length() > 0 && tok2.length() > 0) {
			if (tok1.equals( tok2 )) 
				pair.setFeatureValue( "volumesMatch", 1.0 );
			else
				pair.setFeatureValue( "volumesDontMatch", 1.0 );
		}		
		return carrier;
	}
	
	private String getVenueOrEditionTokens (Citation c) {
		String ret = "";
		ret = c.getField( Citation.volume );
		if (ret.length() > 0) {
			ret = ret.toLowerCase();
			ret = ret.replaceAll( "volume", "" );
			ret = ret.replaceAll( "vol", "" );
			ret = ret.replaceAll( "v ", "" );
			return ret;
		}
		return ret;
	}
}