fix bug on windows

lancopku · Jan 12, 2019 · 4f4aa8a · 4f4aa8a
1 parent 03d0d66
commit 4f4aa8a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/pkuseg/main.py b/pkuseg/main.py
@@ -207,6 +207,7 @@ def basicTrain(config, XTest, tb):
             config.diffList.append(config.diff)
             scoreList = tb.test(XTest, i)
             config.scoreListList.append(scoreList)
+            score = scoreList[0]
 
             logstr = "iter{}  diff={}  train-time(sec)={}  {}={}%".format(
                 config.glbIter,

diff --git a/pkuseg/toolbox.py b/pkuseg/toolbox.py
@@ -259,7 +259,7 @@ def multiThreading(self, X, X2, dynamic=False):
             start = i * interval
             end = min(start + interval, len(X2))
             proc = Process(
-                target=toolbox.taskRunner_test, args=(self, X2, start, end, Q)
+                target=toolbox.taskRunner_test, args=(self.Inf, self.Model, X2, start, end, Q)
             )
             proc.start()
             procs.append(proc)
@@ -271,11 +271,12 @@ def multiThreading(self, X, X2, dynamic=False):
         for proc in procs:
             proc.join()
 
-    def taskRunner_test(self, X2, start, end, Q):
+    @staticmethod
+    def taskRunner_test(Inf, Model, X2, start, end, Q):
         for k in range(start, end):
             x = X2[k]
             tags = []
-            prob = self.Inf.decodeViterbi_fast(self.Model, x._x, tags)
+            prob = Inf.decodeViterbi_fast(Model, x._x, tags)
             Q.put((k, tags))