xueyiming 5 日 前
コミット
3eba9b4574
1 ファイル変更13 行追加0 行削除
  1. 13 0
      src/main/java/examples/utils/SimilarityUtils.java

+ 13 - 0
src/main/java/examples/utils/SimilarityUtils.java

@@ -3,6 +3,9 @@ package examples.utils;
 import com.tzld.piaoquan.recommend.similarity.word2vec.Segment;
 import com.tzld.piaoquan.recommend.similarity.word2vec.Word2Vec;
 import lombok.extern.slf4j.Slf4j;
+import org.ansj.library.AmbiguityLibrary;
+import org.ansj.library.DicLibrary;
+import org.ansj.library.SynonymsLibrary;
 
 import java.io.IOException;
 import java.util.List;
@@ -17,6 +20,7 @@ public final class SimilarityUtils {
     private static Word2Vec vec;
 
     public static void init() {
+        initLibrary();
         Segment.getWords("1");
         try {
             long start = System.currentTimeMillis();
@@ -43,4 +47,13 @@ public final class SimilarityUtils {
         List<String> words2 = Segment.getWords(str2);
         return vec.sentenceSimilarity(words1, words2);
     }
+
+    public static void initLibrary() {
+        AmbiguityLibrary.put(AmbiguityLibrary.DEFAULT, "jar://library/ambiguity.dic");
+        AmbiguityLibrary.reload(AmbiguityLibrary.DEFAULT);
+        DicLibrary.put(DicLibrary.DEFAULT, "jar://library/default.dic");
+        DicLibrary.reload(DicLibrary.DEFAULT);
+        SynonymsLibrary.put(SynonymsLibrary.DEFAULT, "jar://library/synonyms.dic");
+        SynonymsLibrary.reload(SynonymsLibrary.DEFAULT);
+    }
 }