File size: 3,076 Bytes
6cf191b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
#############################
# Imports
#############################
# Python modules
from typing import Tuple, Optional, List
# Remote modules
# Local modules
from .kg_base_wrapper import KGBaseHandler
from utils import read_json_file_2_dict
#############################
# Constants
#############################
#############################
# Handler
#############################
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
|