Browse Source

fix data file import error

chenhaiyang 5 years ago
parent
commit
06576663ef
2 changed files with 15 additions and 3 deletions
  1. 8 2
      sim/text_sim.py
  2. 7 1
      sim/words_sim.py

+ 8 - 2
sim/text_sim.py

@@ -1,13 +1,19 @@
 # -*- coding: utf-8 -*-
 
+import os
 from operator import itemgetter
 
 import jieba
 import numpy as np
-from words_sim import SimCilin
+
+from .words_sim import SimCilin
+
+base_path = os.path.abspath(__file__)
+folder = os.path.dirname(base_path)
+data_path = os.path.join(folder, 'cilin_dict.txt')
 
 ci_lin = SimCilin()
-jieba.load_userdict('cilin_words.txt')
+jieba.load_userdict(data_path)
 
 
 def segmentation(sentence_list):

+ 7 - 1
sim/words_sim.py

@@ -1,13 +1,19 @@
 # -*- coding: utf-8 -*-
 
 import codecs
+import os
 import time
 
+base_path = os.path.abspath(__file__)
+folder = os.path.dirname(base_path)
+data_path = os.path.join(folder, 'cilin_dict.txt')
+
 
 class SimCilin(object):
 
     def __init__(self):
-        self.cilin_path = 'cilin_dict.txt'
+        # self.cilin_path = '/Users/highing/code/sagacloud/cilin/sim/sim/cilin_dict.txt'
+        self.cilin_path = data_path
         self.sem_dict = self.load_semantic()
 
     def load_semantic(self):