public List<TrecTextDocument> getRelevantDocuments(
String timestamp, String workingDirectory, List<Entity> entities, ExecuteQuery eq) {
String query = QueryBuilder.buildOrQuery(getExpansions());
List<TrecTextDocument> docs =
QueryFactory.DoQuery(
Arrays.asList(timestamp), Arrays.asList(query), workingDirectory, entities, eq)
.get(query);
// return docs;
// System.out.println("Originally had " + docs.size() + " docs");
List<TrecTextDocument> disambiguated = disambiguate(docs);
/*
for (TrecTextDocument t : disambiguated)
{
Set<String> relevantEntities = Utils.getManualAnnotationsForDocument(t.docNumber, workingDirectory);
if (!relevantEntities.isEmpty())
{
if (relevantEntities.contains(targetID))
System.out.println("Found document is annotated to be relevant");
else
System.out.println("Found document is annotated as not relevant");
}
}
*/
return disambiguated;
}
public List<TrecTextDocument> getRelevantDocuments(
String timestamp, String workingDirectory, List<Entity> entities) {
String query = QueryBuilder.buildOrQuery(getExpansions());
List<TrecTextDocument> docs =
QueryFactory.DoQuery(
Arrays.asList(timestamp), Arrays.asList(query), workingDirectory, entities, null)
.get(query);
// return docs;
// System.out.println("Originally had " + docs.size() + " docs");
return disambiguate(docs);
}
