org.jbpt.bp.sim.BaselineSimilarity Maven / Gradle / Ivy

Go to download
package org.jbpt.bp.sim;

import org.jbpt.alignment.Alignment;
import org.jbpt.bp.RelSet;
import org.jbpt.hypergraph.abs.IEntity;
import org.jbpt.hypergraph.abs.IEntityModel;
import org.jbpt.petri.Place;
import org.jbpt.petri.Transition;

/**
 * Scores two models by only assessing the overlap of nodes.
 * 
 * @author matthias.weidlich
 *
 */
public class BaselineSimilarity, M extends IEntityModel, N extends IEntity> extends AbstractRelSetSimilarity {

	@Override
	public double score(Alignment alignment) {
		double in1 = 0;
		for (N n : alignment.getFirstModel().getEntities()) {
			if (n instanceof Place) continue;
			if (((Transition)n).isSilent()) continue;
			in1++;
		}
		double in2 = 0;
		for (N n : alignment.getSecondModel().getEntities()) {
			if (n instanceof Place) continue;
			if (((Transition)n).isSilent()) continue;
			in2++;
		}
		
		double intersection = alignment.getAlignedEntitiesOfFirstModel().size();

		return (intersection > 0) ? (intersection / (in1 + in2 - intersection)) : 0;
	}	
	
	@Override
	public double scoreDice(Alignment alignment) {
		double in1 = 0;
		for (N n : alignment.getFirstModel().getEntities()) {
			if (n instanceof Place) continue;
			if (((Transition)n).isSilent()) continue;
			in1++;
		}
		double in2 = 0;
		for (N n : alignment.getSecondModel().getEntities()) {
			if (n instanceof Place) continue;
			if (((Transition)n).isSilent()) continue;
			in2++;
		}
		
		double intersection = alignment.getAlignedEntitiesOfFirstModel().size();

		return (in1 + in2 > 0) ? (2*intersection / (in1 + in2)) : 0;
	}	
}