59 lines
1.5 KiB
Java
59 lines
1.5 KiB
Java
|
|
package eu.dnetlib.pace.tree;
|
|
|
|
import java.util.Map;
|
|
|
|
import com.wcohen.ss.AbstractStringDistance;
|
|
|
|
import eu.dnetlib.pace.config.Config;
|
|
import eu.dnetlib.pace.tree.support.AbstractStringComparator;
|
|
import eu.dnetlib.pace.tree.support.ComparatorClass;
|
|
|
|
/**
|
|
* Compared compare between two titles, ignoring version numbers. Suitable for Software entities.
|
|
*/
|
|
@ComparatorClass("levensteinTitleIgnoreVersion")
|
|
public class LevensteinTitleIgnoreVersion extends AbstractStringComparator {
|
|
|
|
public LevensteinTitleIgnoreVersion(Map<String, String> params) {
|
|
super(params, new com.wcohen.ss.Levenstein());
|
|
}
|
|
|
|
public LevensteinTitleIgnoreVersion(final double w) {
|
|
super(w, new com.wcohen.ss.Levenstein());
|
|
}
|
|
|
|
protected LevensteinTitleIgnoreVersion(final double w, final AbstractStringDistance ssalgo) {
|
|
super(w, ssalgo);
|
|
}
|
|
|
|
@Override
|
|
public double distance(final String a, final String b, final Config conf) {
|
|
String ca = cleanup(a);
|
|
String cb = cleanup(b);
|
|
|
|
ca = ca.replaceAll("\\d", "").replaceAll(getRomans(ca), "").trim();
|
|
cb = cb.replaceAll("\\d", "").replaceAll(getRomans(cb), "").trim();
|
|
|
|
ca = filterAllStopWords(ca);
|
|
cb = filterAllStopWords(cb);
|
|
|
|
return normalize(ssalgo.score(ca, cb), ca.length(), cb.length());
|
|
}
|
|
|
|
private double normalize(final double score, final int la, final int lb) {
|
|
return 1 - (Math.abs(score) / Math.max(la, lb));
|
|
}
|
|
|
|
@Override
|
|
public double getWeight() {
|
|
return super.weight;
|
|
}
|
|
|
|
@Override
|
|
protected double normalize(final double d) {
|
|
return 1 / Math.pow(Math.abs(d) + 1, 0.1);
|
|
}
|
|
|
|
}
|