diff --git a/jiagu/segment/dict/user.dict b/jiagu/segment/dict/user.dict index 556bad6..af694f4 100644 --- a/jiagu/segment/dict/user.dict +++ b/jiagu/segment/dict/user.dict @@ -1,2 +1,3 @@ 思知 - +天猫 +知识图谱 diff --git a/jiagu/segment/nroute.py b/jiagu/segment/nroute.py index 3184b08..09dd043 100644 --- a/jiagu/segment/nroute.py +++ b/jiagu/segment/nroute.py @@ -15,8 +15,9 @@ class Segment: self.total_freq = 0 self.initialized = False - def init(self, vocab_path='dict/jiagu.dict'): + def init(self, vocab_path='dict/jiagu.dict', user_vocab='dict/user.dict'): self.load_vocab(os.path.join(os.path.dirname(__file__), vocab_path)) + self.load_vocab(os.path.join(os.path.dirname(__file__), user_vocab)) self.initialized = True def load_vocab(self, vocab_path): @@ -179,9 +180,9 @@ if __name__=='__main__': s = Segment() # sg.load_userdict('dict/user.dict') - s.load_userdict(['知识图谱']) + # s.load_userdict(['知识图谱']) - text = '知识图谱机器人hello\nworld¥¥' + text = '这一天猫小黑盒' words = s.seg(text) print(words) @@ -191,4 +192,4 @@ if __name__=='__main__': - \ No newline at end of file +