修复word2vec缓存问题：#718

hankcs · Dec 19, 2017 · e704dae · e704dae
1 parent 28a790c
commit e704dae
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/main/java/com/hankcs/hanlp/mining/word2vec/TextFileCorpus.java b/src/main/java/com/hankcs/hanlp/mining/word2vec/TextFileCorpus.java
@@ -66,7 +66,13 @@ void reduceVocab()
             DataInputStream oldCache = new DataInputStream(new FileInputStream(cacheFile));
             while (oldCache.available() >= 4)
             {
-                int id = table[oldCache.readInt()];
+                int oldId = oldCache.readInt();
+                if (oldId < 0)
+                {
+                    cache.writeInt(oldId);
+                    continue;
+                }
+                int id = table[oldId];
                 if (id == -4) continue;
                 cache.writeInt(id);
             }