fastText: fix FastText#loadInput (issue #1)

sszuev · May 2, 2018 · 160c901 · 160c901
1 parent fce0391
commit 160c901
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/main/java/cc/fasttext/FastText.java b/src/main/java/cc/fasttext/FastText.java
@@ -1379,6 +1379,7 @@ protected Matrix loadInput(Args args, Dictionary dictionary, String file) throws
                     if (numbers.size() < dim)
                         throw new IllegalArgumentException("Wrong numbers in the line: " + numbers.size() + ". Expected " + dim);
                     words.add(word);
+                    dictionary.add(word);
                     for (int j = 0; j < numbers.size(); j++) {
                         mat.set(i, j, numbers.get(j));
                     }