提高DocVectorModel健壮性

Author: hankcs

src/main/java/com/hankcs/hanlp/mining/word2vec/DocVectorModel.java

@@ -68,17 +68,23 @@ public Vector query(String content)
     {
         if (content == null || content.length() == 0) return null;
         List<Term> termList = NotionalTokenizer.segment(content);
-        if (termList.isEmpty())
-        {
-            return null;
-        }
         Vector result = new Vector(dimension());
+        int n = 0;
         for (Term term : termList)
         {
             Vector vector = wordVectorModel.vector(term.word);
+            if (vector == null)
+            {
+                continue;
+            }
+            ++n;
             result.addToSelf(vector);
         }
-        result.divideToSelf(termList.size());
+        if (n == 0)
+        {
+            return null;
+        }
+        result.divideToSelf(n);
         return result;
     }