|
@@ -3,6 +3,9 @@ package examples.utils;
|
|
import com.tzld.piaoquan.recommend.similarity.word2vec.Segment;
|
|
import com.tzld.piaoquan.recommend.similarity.word2vec.Segment;
|
|
import com.tzld.piaoquan.recommend.similarity.word2vec.Word2Vec;
|
|
import com.tzld.piaoquan.recommend.similarity.word2vec.Word2Vec;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
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.io.IOException;
|
|
import java.util.List;
|
|
import java.util.List;
|
|
@@ -17,6 +20,7 @@ public final class SimilarityUtils {
|
|
private static Word2Vec vec;
|
|
private static Word2Vec vec;
|
|
|
|
|
|
public static void init() {
|
|
public static void init() {
|
|
|
|
+ initLibrary();
|
|
Segment.getWords("1");
|
|
Segment.getWords("1");
|
|
try {
|
|
try {
|
|
long start = System.currentTimeMillis();
|
|
long start = System.currentTimeMillis();
|
|
@@ -43,4 +47,13 @@ public final class SimilarityUtils {
|
|
List<String> words2 = Segment.getWords(str2);
|
|
List<String> words2 = Segment.getWords(str2);
|
|
return vec.sentenceSimilarity(words1, words2);
|
|
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);
|
|
|
|
+ }
|
|
}
|
|
}
|