it.unipi.di.acube.batframework.metrics.NaiveScoredTagMatch Maven / Gradle / Ivy
/**
* (C) Copyright 2012-2013 A-cube lab - Università di Pisa - Dipartimento di Informatica.
* BAT-Framework is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
* BAT-Framework is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
* You should have received a copy of the GNU General Public License along with BAT-Framework. If not, see .
*/
package it.unipi.di.acube.batframework.metrics;
import it.unipi.di.acube.batframework.data.*;
import it.unipi.di.acube.batframework.utils.WikipediaApiInterface;
import java.io.IOException;
import java.util.*;
public class NaiveScoredTagMatch implements MatchRelation{
private WikipediaApiInterface api;
public NaiveScoredTagMatch(WikipediaApiInterface api){
this.api = api;
}
@Override
public boolean match(ScoredTag t1, ScoredTag t2) {
try {
return api.dereference(t1.getConcept()) == api.dereference(t2.getConcept());
} catch (IOException e) {
e.printStackTrace();
throw new RuntimeException(e);
}
}
@Override
public List> preProcessOutput(List> computedOutput) {
try {
Annotation.prefetchRedirectList(computedOutput, api);
} catch (IOException e) {
e.printStackTrace();
throw new RuntimeException(e);
}
return computedOutput;
}
@Override
public List> preProcessGoldStandard(List> goldStandard) {
return preProcessOutput(goldStandard);
}
@Override
public String getName() {
return "Naive scored tag match";
}
}