fixed a bug related to writing the word vector to the file

Author: hardikp

glove.py

@@ -86,7 +86,7 @@ def train_glove(args):
         j = Variable(torch.LongTensor([i]))
         vec = (model.l_vecs(j) + model.r_vecs(j)).data.squeeze()
         for k in range(vec.size(0)):
-            s + ' ' + str(vec[k])
+            s += ' ' + str(round(vec[k], 6))
 
         f.write(s)
 

test_word_vectors.py

@@ -7,7 +7,7 @@
 
 
 def print_nearest_words(args):
-    word = args.word.lower()
+    word = args.word.lower().strip()
 
     # Load the word vectors
     embeddings_index = {}
@@ -19,8 +19,8 @@ def print_nearest_words(args):
         embeddings_index[w] = coefs
     f.close()
 
-    w_v = np.zeros(50)
-    for w in word.strip().split():
+    w_v = np.zeros_like(embeddings_index[w])
+    for w in word.split():
         if w not in embeddings_index.keys():
             continue