fix add_sentence_table third times

code/prediction.py

@@ -79,23 +79,15 @@ def add_sentence_table(result, pattern =":；。，？!～！： ", threshold_nu
     sample["predict_sentence_table"] = torch.zeros(len(sample["predict_span_table"]))
     for boundary in boundary_list:
       end = boundary
-      new_start = start
-      new_end = end
-      predict_num = 0
-      for i, predict_char in enumerate(sample["predict_span_table"][start:end]):
-        if predict_char!=0:
-          predict_num+=1
+      predict_num = sum(sample["predict_span_table"][start:end]>0)
       sentence_num = len(sample["predict_span_table"][start:end])
       if(predict_num > threshold_num) or (predict_num > sentence_num*threshold_rate):
-        
-        while(sample["predict_span_table"][new_start]==0): new_start+=1
-        while(sample["predict_span_table"][new_end-1]==0): new_end-=1
         if fist_sentence:
-          sample["predict_sentence_table"][new_start:new_end] = 2
-          sample["predict_sentence_table"][new_start] = 1
+          sample["predict_sentence_table"][start:end] = 2
+          sample["predict_sentence_table"][start] = 1
+          fist_sentence=False
         else: 
-          sample["predict_sentence_table"][new_start-1:new_end] = 2
-        fist_sentence=False
+          sample["predict_sentence_table"][start-1:end] = 2
       else: fist_sentence =True
       start = end+1