rogel 4 жил өмнө
parent
commit
093b4631ae

+ 0 - 2
BiddingKG/dl/interface/Preprocessing.py

@@ -1917,8 +1917,6 @@ def union_result(codeName,prem):
     @return:拼接起来的字典
     '''
     result = []
-    print(codeName)
-    print(prem)
     assert len(codeName)==len(prem)
     for item_code,item_prem in zip(codeName,prem):
         result.append(dict(item_code,**item_prem))

+ 2 - 1
BiddingKG/dl/test/test4.py

@@ -52,7 +52,7 @@ epcPredict = predictor.EPCPredict()
 roleRulePredict = predictor.RoleRulePredictor()
 timePredict = predictor.TimePredictor()
 punish = punish_rule.Punish_Extract()
-
+productPredict = predictor.ProductPredictor()
 
 #自定义jsonEncoder
 class MyEncoder(json.JSONEncoder):
@@ -80,6 +80,7 @@ def predict(doc_id,text,title=""):
     codeName = codeNamePredict.predict(list_sentences,list_entitys=list_entitys)
     print(codeName)
     premPredict.predict(list_sentences,list_entitys)
+    productPredict.predict(list_sentences,list_entitys)
     # roleRulePredict.predict(list_articles,list_sentences, list_entitys,codeName)
     print("epcPredict")
     epcPredict.predict(list_sentences,list_entitys)