def test_wiki_online():
mentions = get_wiki_mentions()
wiki = WikipediaRelationExtraction()
assert not wiki.extract_all_relations(mentions[0], mentions[0]).isdisjoint(
set(
[
RelationType.WIKIPEDIA_CATEGORY,
RelationType.WIKIPEDIA_REDIRECT_LINK,
RelationType.WIKIPEDIA_BE_COMP,
]
)
)
assert not wiki.extract_all_relations(mentions[0], mentions[1]).isdisjoint(
set(
[
RelationType.WIKIPEDIA_CATEGORY,
RelationType.WIKIPEDIA_REDIRECT_LINK,
RelationType.WIKIPEDIA_BE_COMP,
]
)
)
assert (
wiki.extract_all_relations(mentions[0], mentions[2]).pop() == RelationType.NO_RELATION_FOUND
)
def run_example():
logger.info("Running relation extraction example......")
computed = ComputedRelationExtraction()
ref_dict = ReferentDictRelationExtraction(
ref_dict=str(LIBRARY_ROOT / "datasets" / "coref.dict1.tsv"))
vo = VerboceanRelationExtraction(
vo_file=str(LIBRARY_ROOT / "datasets" /
"verbocean.unrefined.2004-05-20.txt"))
wiki = WikipediaRelationExtraction()
wn = WordnetRelationExtraction()
embed = WordEmbeddingRelationExtraction(method=EmbeddingMethod.ELMO)
mention_x1 = MentionDataLight(
"IBM",
mention_context=
"IBM manufactures and markets computer hardware, middleware and software",
)
mention_y1 = MentionDataLight(
"International Business Machines",
mention_context="International Business Machines Corporation is an "
"American multinational information technology company",
)
computed_relations = computed.extract_all_relations(mention_x1, mention_y1)
ref_dict_relations = ref_dict.extract_all_relations(mention_x1, mention_y1)
vo_relations = vo.extract_all_relations(mention_x1, mention_y1)
wiki_relations = wiki.extract_all_relations(mention_x1, mention_y1)
embed_relations = embed.extract_all_relations(mention_x1, mention_y1)
wn_relaions = wn.extract_all_relations(mention_x1, mention_y1)
if RelationType.NO_RELATION_FOUND in computed_relations:
logger.info("No Computed relation found")
else:
logger.info("Found Computed relations-%s",
str(list(computed_relations)))
if RelationType.NO_RELATION_FOUND in ref_dict_relations:
logger.info("No Referent-Dict relation found")
else:
logger.info("Found Referent-Dict relations-%s",
str(list(ref_dict_relations)))
if RelationType.NO_RELATION_FOUND in vo_relations:
logger.info("No Verb-Ocean relation found")
else:
logger.info("Found Verb-Ocean relations-%s", str(list(vo_relations)))
if RelationType.NO_RELATION_FOUND in wiki_relations:
logger.info("No Wikipedia relation found")
else:
logger.info("Found Wikipedia relations-%s", str(wiki_relations))
if RelationType.NO_RELATION_FOUND in embed_relations:
logger.info("No Embedded relation found")
else:
logger.info("Found Embedded relations-%s", str(list(embed_relations)))
if RelationType.NO_RELATION_FOUND in wn_relaions:
logger.info("No Wordnet relation found")
else:
logger.info("Found Wordnet relations-%s", str(wn_relaions))