Merge pull request #423 from jacquesqiao/fix-word2vec-load

fix word2vec load

04.word2vec/README.cn.md

@@ -223,7 +223,7 @@ def load_dict_and_embedding():
     with open("word_dict", "r") as f:
         for line in f:
             key, value = line.strip().split(" ")
-            word_dict[key] = value
+            word_dict[key] = int(value)
 
     embeddings = numpy.loadtxt("embedding_table", delimiter=",")
     return word_dict, embeddings

04.word2vec/README.md

@@ -241,7 +241,7 @@ def load_dict_and_embedding():
     with open("word_dict", "r") as f:
         for line in f:
             key, value = line.strip().split(" ")
-            word_dict[key] = value
+            word_dict[key] = int(value)
 
     embeddings = numpy.loadtxt("embedding_table", delimiter=",")
     return word_dict, embeddings

04.word2vec/index.cn.html

@@ -265,7 +265,7 @@ def load_dict_and_embedding():
     with open("word_dict", "r") as f:
         for line in f:
             key, value = line.strip().split(" ")
-            word_dict[key] = value
+            word_dict[key] = int(value)
 
     embeddings = numpy.loadtxt("embedding_table", delimiter=",")
     return word_dict, embeddings

04.word2vec/index.html

@@ -283,7 +283,7 @@ def load_dict_and_embedding():
     with open("word_dict", "r") as f:
         for line in f:
             key, value = line.strip().split(" ")
-            word_dict[key] = value
+            word_dict[key] = int(value)
 
     embeddings = numpy.loadtxt("embedding_table", delimiter=",")
     return word_dict, embeddings

04.word2vec/train.py

@@ -34,7 +34,7 @@ def load_dict_and_embedding():
     with open("word_dict", "r") as f:
         for line in f:
             key, value = line.strip().split(" ")
-            word_dict[key] = value
+            word_dict[key] = int(value)
 
     embeddings = numpy.loadtxt("embedding_table", delimiter=",")
     return word_dict, embeddings