소스 검색

Merge branch 'master' of http://192.168.2.103:3000/luojiehua/BIDI_ML_INFO_EXTRACTION

znj 6 달 전
부모
커밋
ebc1a58ed7
1개의 변경된 파일3개의 추가작업 그리고 0개의 파일을 삭제
  1. 3 0
      BiddingKG/dl/interface/extract.py

+ 3 - 0
BiddingKG/dl/interface/extract.py

@@ -285,6 +285,9 @@ def predict(doc_id,text,title="",page_time="",web_source_no='',web_source_name="
 
     start_time = time.time() # 项目编号、名称提取
     codeName = predictor.getPredictor("codeName").predict(list_sentences,MAX_AREA=5000,list_entitys=list_entitys)
+    if re.search('破产清算案', title):
+        end = re.search('破产清算案', title).end()
+        codeName[0]['name'] = title[:end]
     log("get codename done of doc_id%s"%(doc_id))
     cost_time["codename"] = round(time.time()-start_time,2)