Gensim Test

gensim-test.py

"""
Tests gensim with the Google News Word2Vec model
"""

import gensim
import numpy
from scipy import spatial

M = gensim.models.KeyedVectors.load_word2vec_format(
    "./GoogleNews-vectors-negative300.bin", binary=True
)

while True:
    try: 
        vec1 = numpy.zeros(300)
        for w in raw_input("Word 1: ").split(" "):
            vec1 = vec1 + M.wv[w]

        vec2 = numpy.zeros(300)
        for w in raw_input("Word 2: ").split(" "):
            vec2 = vec2 + M.wv[w]

        print(1 - spatial.distance.cosine(vec1, vec2))
    except Exception as e:
        print(e.message)