#!/usr/bin/env python

import gensim

print("Loading Word Vectors...")
model = gensim.models.KeyedVectors.load_word2vec_format('~/data/bio_nlp_vec/PubMed-shuffle-win-2.bin', binary=True)
print("done.")

while(True):
        utterance = input("Enter two words: ")
        if utterance == "stop":
            break
        word1, word2 = utterance.split(",").strip()

        print("{0}: {1}".format(word1, sum(model[word1])))
        print("{0}: {1}".format(word2, sum(model[word2])))
