|
|
|
|
|
|
|
|
|
|
|
from typing import Tuple, Optional, List |
|
|
|
|
|
|
|
from .kg_base_wrapper import KGBaseHandler |
|
from utils import read_json_file_2_dict |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class ConceptNetHandler(KGBaseHandler): |
|
def __init__(self, database=""): |
|
super(ConceptNetHandler, self).__init__() |
|
_store_dir = 'kgs_binding/conceptnet' |
|
self.conceptnet_concepts = read_json_file_2_dict('conceptnet_english_nouns_simple.json', store_dir=_store_dir) |
|
self.relations_concepts = read_json_file_2_dict('conceptnet_english_noun_2_noun_relations.json', store_dir=_store_dir) |
|
self.concept_2_concepts = read_json_file_2_dict('conceptnet_english_nouns.json', store_dir=_store_dir) |
|
|
|
def get_relation_types(self) -> List[str]: |
|
updated_relation_names = ['not_has_property', 'not_desires', 'external_u_r_l', 'created_by', |
|
'not_capable_of', 'antonym', 'has_first_subevent', 'located_near', |
|
'desires', 'has_prerequisite', 'has_last_subevent', 'synonym', 'is_a', |
|
'manner_of', 'has_a', 'motivated_by_goal', 'instance_of', |
|
'etymologically_derived_from', 'capable_of', 'for', 'at_location', |
|
'has_subevent', 'causes', 'has_context', 'symbol_of', 'derived_from', |
|
'made_of', 'causes_desire', 'has_property', 'similar_to', 'used_for', 'by', |
|
'entails', 'form_of', 'receives_action', 'distinct_from', 'related_to', |
|
'part_of', 'defined_as', 'etymologically_related_to'] |
|
return updated_relation_names |
|
|
|
def exists_relation_between(self, concept, other_concept) -> bool: |
|
left_2_right, right_2_left = self.relation_between(concept, other_concept) |
|
return left_2_right is not None or right_2_left is not None |
|
|
|
def relation_between(self, concept, other_concept) -> Tuple[Optional[str], Optional[str]]: |
|
left_2_right_txt = f'{concept}|{other_concept}' |
|
right_2_left_txt = f'{other_concept}|{concept}' |
|
left_2_right_relations = self.relations_concepts.get(left_2_right_txt, None) |
|
right_2_left_relations = self.relations_concepts.get(right_2_left_txt, None) |
|
left_2_right_relation, right_2_left_relation = None, None |
|
if left_2_right_relations: |
|
left_2_right_relation = self.ignore_less_relevant_connection(left_2_right_relations) |
|
if right_2_left_relations: |
|
right_2_left_relation = self.ignore_less_relevant_connection(right_2_left_relations) |
|
return left_2_right_relation, right_2_left_relation |
|
|
|
def get_related_concepts(self, concept) -> Optional[List[str]]: |
|
return self.concept_2_concepts.get(concept, []) |
|
|
|
def does_concept_exist(self, concept) -> bool: |
|
return concept in self.conceptnet_concepts |
|
|
