维特比分词设置默认自定义词典

TimerYue · Dec 3, 2018 · 4b03c2b · 4b03c2b
1 parent 4a29ae0
commit 4b03c2b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/main/java/com/hankcs/hanlp/seg/Viterbi/ExtendViterbiSegment.java b/src/main/java/com/hankcs/hanlp/seg/Viterbi/ExtendViterbiSegment.java
@@ -24,7 +24,6 @@ public class ExtendViterbiSegment extends ViterbiSegment
     public ExtendViterbiSegment()
     {
         super();
-        super.setDat(CustomDictionary.dat);
     }
 
     /**
@@ -100,7 +99,6 @@ private void loadCustomDic(String customPath, boolean cache)
             if (map.size() == 0)
             {
                 logger.warning("没有加载到任何词条，使用默认词典");
-                super.setDat(CustomDictionary.dat);
                 return;
             }
             logger.info("正在构建DoubleArrayTrie……");
@@ -139,7 +137,6 @@ private void loadCustomDic(String customPath, boolean cache)
         catch (Exception e)
         {
             logger.warning("自定义词典" + customPath + "缓存失败！\n" + TextUtility.exceptionToString(e));
-            super.setDat(CustomDictionary.dat);
         }
     }
 }
diff --git a/src/main/java/com/hankcs/hanlp/seg/Viterbi/ViterbiSegment.java b/src/main/java/com/hankcs/hanlp/seg/Viterbi/ViterbiSegment.java
@@ -14,6 +14,7 @@
 import com.hankcs.hanlp.HanLP;
 import com.hankcs.hanlp.collection.trie.DoubleArrayTrie;
 import com.hankcs.hanlp.dictionary.CoreDictionary;
+import com.hankcs.hanlp.dictionary.CustomDictionary;
 import com.hankcs.hanlp.recognition.nr.JapanesePersonRecognition;
 import com.hankcs.hanlp.recognition.nr.PersonRecognition;
 import com.hankcs.hanlp.recognition.nr.TranslatedPersonRecognition;
@@ -37,6 +38,11 @@ public class ViterbiSegment extends WordBasedSegment
 {
     private DoubleArrayTrie<CoreDictionary.Attribute> dat = new DoubleArrayTrie<CoreDictionary.Attribute>();
 
+    public ViterbiSegment()
+    {
+        this.dat = CustomDictionary.dat;
+    }
+
     public DoubleArrayTrie<CoreDictionary.Attribute> getDat()
     {
         return dat;