def annotate(self, doc, doc_id):
"""Performs entity linking and annotates the query."""
mention_ens = self.parse(doc, doc_id)
ELUtils.debug_mention_detection(mention_ens)
disamb_ens = self.disambiguate(mention_ens)
ELUtils.debug_disambiguation(disamb_ens)
return disamb_ens
def main():
snippets = Document.load_test_snippets(SNIPPETS)
out_file_name = OUTPUT_DIR + "/output_cmn.txt"
out_file = open(out_file_name, "w")
el = ELCmn()
for doc_id, doc in sorted(snippets.items(), key=lambda item: int(item[0])):
print("[" + doc_id + "]\t" + doc)
linked_ens = el.annotate(doc, doc_id)
ELUtils.write_to_file(doc_id, out_file, linked_ens)
print("------")
print("output written to", out_file_name)