补提交时间分类pb模型文件及更新test4.py

BiddingKG/dl/test/test4.py

@@ -92,18 +92,27 @@ def predict(doc_id,text):
     punish_dic = punish.get_punish_extracts(list_sentences, list_entitys, title='投诉处理 ', text=text)
     print(punish_dic)
     prem[0][1]['punish'] = punish_dic
-    # 招标方式
-    bidway = [entity.entity_text for entity in list_entitys[0] if entity.entity_type=='bidway']
-    # 资金来源
-    moneySource = [entity.entity_text for entity in list_entitys[0] if entity.entity_type=='moneySource']
-    # 服务时间
-    servicetime = [entity.entity_text for entity in list_entitys[0] if entity.entity_type=='servicetime']
-    # 发布时间 time_release：1
-    time_release = [entity.entity_text for entity in list_entitys[0] if entity.entity_type == 'time' and entity.label==1]
-    # 开标时间  'time_bidopen':2,
-    time_bidopen = [entity.entity_text for entity in list_entitys[0] if entity.entity_type == 'time' and entity.label==2]
-    # 截标时间 'time_bidclose':3
-    time_bidclose = [entity.entity_text for entity in list_entitys[0] if entity.entity_type == 'time' and entity.label == 3]
+
+    bidway = [] # 招标方式
+    moneySource = [] # 资金来源
+    servicetime = [] # 服务时间
+    time_release = [] # 发布时间
+    time_bidopen = [] # 开标时间
+    time_bidclose = [] # 截标时间
+    for entity in list_entitys[0]:
+        if entity.entity_type == 'bidway':
+            bidway.append(entity.entity_text)
+        elif entity.entity_type=='moneySource':
+            moneySource.append(entity.entity_text)
+        elif entity.entity_type=='servicetime':
+            servicetime.append(entity.entity_text)
+        elif entity.entity_type == 'time' and entity.label==1:
+            time_release.append(entity.entity_text)
+        elif entity.entity_type == 'time' and entity.label==2:
+            time_bidopen.append(entity.entity_text)
+        elif entity.entity_type == 'time' and entity.label == 3:
+            time_bidclose.append(entity.entity_text)
+
     prem[0][1]['bidway'] = '；'.join(set(bidway))
     prem[0][1]['moneySource'] = '；'.join(set(moneySource))
     prem[0][1]['servicetime'] = '；'.join(set(servicetime))