nestedcrf.java

来自「CRF1.2」· Java 代码 · 共 40 行

package iitb.CRF;

/**
 *
 * @author Sunita Sarawagi
 *
 */ 
public class NestedCRF extends CRF {
    FeatureGeneratorNested featureGenNested;
    transient NestedViterbi nestedViterbi;
    public NestedCRF(int numLabels, FeatureGeneratorNested fgen, String arg) {
	super(numLabels,fgen,arg);
	featureGenNested = fgen;
	nestedViterbi = new NestedViterbi(this,1);
    }
    public NestedCRF(int numLabels, FeatureGeneratorNested fgen, java.util.Properties configOptions) {
	super(numLabels,fgen,configOptions);
	featureGenNested = fgen;
	nestedViterbi = new NestedViterbi(this,1);
    }
    protected Trainer getTrainer() {
	if (params.trainerType.startsWith("SegmentCollins"))
	    return new NestedCollinsTrainer(params);
	return new NestedTrainer(params);
    }
    protected Viterbi getViterbi(int beamsize) {
    	return new NestedViterbi(this,beamsize);
    }
    public void apply(DataSequence dataSeq) {
    	apply((SegmentDataSequence)dataSeq);
    }
    public void apply(SegmentDataSequence dataSeq) {
        if (nestedViterbi==null)
            nestedViterbi = new NestedViterbi(this,1);
	if (params.debugLvl > 2) 
	    Util.printDbg("NestedCRF: Applying on " + dataSeq);
	nestedViterbi.bestLabelSequence(dataSeq,lambda);
    }
};