jonas/nlp-lab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixing relative filepath search of doc2vec model in naive approach

Browse Source
This commit is contained in:
Jonas Weinz 2018-07-19 13:26:07 +02:00
parent da0b23ddeb
commit 406460d031
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Project/naive_approach/naive_approach.py
View File

@ -43,7 +43,8 @@ def stemming(message):
# * compare words to emoji descriptions # * compare words to emoji descriptions
def evaluate_sentence(sentence, description_key = 'description', lang = 'eng', emojis_to_consider="all", stem=True): def evaluate_sentence(sentence, description_key = 'description', lang = 'eng', emojis_to_consider="all", stem=True):
# assumes there is a trained w2v model stored in the same directory! # assumes there is a trained w2v model stored in the same directory!
wv = KeyedVectors.load("word2vec.model", mmap='r') __location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
wv = KeyedVectors.load(str(__location__)+"/word2vec.model", mmap='r')
if (stem): if (stem):
sentence = stemming(sentence) sentence = stemming(sentence)