Utils.py 66 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493
  1. '''
  2. Created on 2018年12月20日
  3. @author: User
  4. '''
  5. import numpy as np
  6. import re
  7. import gensim
  8. from keras import backend as K
  9. import os,sys
  10. import time
  11. import traceback
  12. from threading import RLock
  13. # from pai_tf_predict_proto import tf_predict_pb2
  14. import requests
  15. model_w2v = None
  16. lock_model_w2v = RLock()
  17. USE_PAI_EAS = False
  18. Lazy_load = False
  19. # API_URL = "http://192.168.2.103:8802"
  20. API_URL = "http://127.0.0.1:888"
  21. # USE_API = True
  22. USE_API = False
  23. def getCurrent_date(format="%Y-%m-%d %H:%M:%S"):
  24. _time = time.strftime(format,time.localtime())
  25. return _time
  26. def getw2vfilepath():
  27. filename = "wiki_128_word_embedding_new.vector"
  28. w2vfile = getFileFromSysPath(filename)
  29. if w2vfile is not None:
  30. return w2vfile
  31. return filename
  32. def getLazyLoad():
  33. global Lazy_load
  34. return Lazy_load
  35. def getFileFromSysPath(filename):
  36. for _path in sys.path:
  37. if os.path.isdir(_path):
  38. for _file in os.listdir(_path):
  39. _abspath = os.path.join(_path,_file)
  40. if os.path.isfile(_abspath):
  41. if _file==filename:
  42. return _abspath
  43. return None
  44. model_word_file = os.path.dirname(__file__)+"/../singlew2v_model.vector"
  45. model_word = None
  46. lock_model_word = RLock()
  47. from decimal import Decimal
  48. import logging
  49. logging.basicConfig(level = logging.INFO,format = '%(asctime)s - %(name)s - %(levelname)s - %(message)s')
  50. logger = logging.getLogger(__name__)
  51. import pickle
  52. import os
  53. import json
  54. #自定义jsonEncoder
  55. class MyEncoder(json.JSONEncoder):
  56. def __init__(self):
  57. import numpy as np
  58. global np
  59. def default(self, obj):
  60. if isinstance(obj, np.ndarray):
  61. return obj.tolist()
  62. elif isinstance(obj, bytes):
  63. return str(obj, encoding='utf-8')
  64. elif isinstance(obj, (np.float_, np.float16, np.float32,
  65. np.float64)):
  66. return float(obj)
  67. elif isinstance(obj,(np.int64,np.int32)):
  68. return int(obj)
  69. return json.JSONEncoder.default(self, obj)
  70. vocab_word = None
  71. vocab_words = None
  72. file_vocab_word = "vocab_word.pk"
  73. file_vocab_words = "vocab_words.pk"
  74. selffool_authorization = "NjlhMWFjMjVmNWYyNzI0MjY1OGQ1M2Y0ZmY4ZGY0Mzg3Yjc2MTVjYg=="
  75. selffool_url = "http://pai-eas-vpc.cn-beijing.aliyuncs.com/api/predict/selffool_gpu"
  76. selffool_seg_authorization = "OWUwM2Q0ZmE3YjYxNzU4YzFiMjliNGVkMTA3MzJkNjQ2MzJiYzBhZg=="
  77. selffool_seg_url = "http://pai-eas-vpc.cn-beijing.aliyuncs.com/api/predict/selffool_seg_gpu"
  78. codename_authorization = "Y2M5MDUxMzU1MTU4OGM3ZDk2ZmEzYjkxYmYyYzJiZmUyYTgwYTg5NA=="
  79. codename_url = "http://pai-eas-vpc.cn-beijing.aliyuncs.com/api/predict/codename_gpu"
  80. form_item_authorization = "ODdkZWY1YWY0NmNhNjU2OTI2NWY4YmUyM2ZlMDg1NTZjOWRkYTVjMw=="
  81. form_item_url = "http://pai-eas-vpc.cn-beijing.aliyuncs.com/api/predict/form"
  82. person_authorization = "N2I2MDU2N2Q2MGQ0ZWZlZGM3NDkyNTA1Nzc4YmM5OTlhY2MxZGU1Mw=="
  83. person_url = "http://pai-eas-vpc.cn-beijing.aliyuncs.com/api/predict/person"
  84. role_authorization = "OWM1ZDg5ZDEwYTEwYWI4OGNjYmRlMmQ1NzYwNWNlZGZkZmRmMjE4OQ=="
  85. role_url = "http://pai-eas-vpc.cn-beijing.aliyuncs.com/api/predict/role"
  86. money_authorization = "MDQyNjc2ZDczYjBhYmM4Yzc4ZGI4YjRmMjc3NGI5NTdlNzJiY2IwZA=="
  87. money_url = "http://pai-eas-vpc.cn-beijing.aliyuncs.com/api/predict/money"
  88. codeclasses_authorization = "MmUyNWIxZjQ2NjAzMWJlMGIzYzkxMjMzNWY5OWI3NzJlMWQ1ZjY4Yw=="
  89. codeclasses_url = "http://pai-eas-vpc.cn-beijing.aliyuncs.com/api/predict/codeclasses"
  90. def viterbi_decode(score, transition_params):
  91. """Decode the highest scoring sequence of tags outside of TensorFlow.
  92. This should only be used at test time.
  93. Args:
  94. score: A [seq_len, num_tags] matrix of unary potentials.
  95. transition_params: A [num_tags, num_tags] matrix of binary potentials.
  96. Returns:
  97. viterbi: A [seq_len] list of integers containing the highest scoring tag
  98. indices.
  99. viterbi_score: A float containing the score for the Viterbi sequence.
  100. """
  101. trellis = np.zeros_like(score)
  102. backpointers = np.zeros_like(score, dtype=np.int32)
  103. trellis[0] = score[0]
  104. for t in range(1, score.shape[0]):
  105. v = np.expand_dims(trellis[t - 1], 1) + transition_params
  106. trellis[t] = score[t] + np.max(v, 0)
  107. backpointers[t] = np.argmax(v, 0)
  108. viterbi = [np.argmax(trellis[-1])]
  109. for bp in reversed(backpointers[1:]):
  110. viterbi.append(bp[viterbi[-1]])
  111. viterbi.reverse()
  112. viterbi_score = np.max(trellis[-1])
  113. return viterbi, viterbi_score
  114. def limitRun(sess,list_output,feed_dict,MAX_BATCH=1024):
  115. len_sample = 0
  116. if len(feed_dict.keys())>0:
  117. len_sample = len(feed_dict[list(feed_dict.keys())[0]])
  118. if len_sample>MAX_BATCH:
  119. list_result = [[] for _ in range(len(list_output))]
  120. _begin = 0
  121. while(_begin<len_sample):
  122. new_dict = dict()
  123. for _key in feed_dict.keys():
  124. if isinstance(feed_dict[_key],(float,int,np.int32,np.float_,np.float16,np.float32,np.float64)):
  125. new_dict[_key] = feed_dict[_key]
  126. else:
  127. new_dict[_key] = feed_dict[_key][_begin:_begin+MAX_BATCH]
  128. _output = sess.run(list_output,feed_dict=new_dict)
  129. for _index in range(len(list_output)):
  130. list_result[_index].extend(_output[_index])
  131. _begin += MAX_BATCH
  132. else:
  133. list_result = sess.run(list_output,feed_dict=feed_dict)
  134. return list_result
  135. def get_values(response,output_name):
  136. """
  137. Get the value of a specified output tensor
  138. :param output_name: name of the output tensor
  139. :return: the content of the output tensor
  140. """
  141. output = response.outputs[output_name]
  142. if output.dtype == tf_predict_pb2.DT_FLOAT:
  143. _value = output.float_val
  144. elif output.dtype == tf_predict_pb2.DT_INT8 or output.dtype == tf_predict_pb2.DT_INT16 or \
  145. output.dtype == tf_predict_pb2.DT_INT32:
  146. _value = output.int_val
  147. elif output.dtype == tf_predict_pb2.DT_INT64:
  148. _value = output.int64_val
  149. elif output.dtype == tf_predict_pb2.DT_DOUBLE:
  150. _value = output.double_val
  151. elif output.dtype == tf_predict_pb2.DT_STRING:
  152. _value = output.string_val
  153. elif output.dtype == tf_predict_pb2.DT_BOOL:
  154. _value = output.bool_val
  155. return np.array(_value).reshape(response.outputs[output_name].array_shape.dim)
  156. def vpc_requests(url,authorization,request_data,list_outputs):
  157. headers = {"Authorization": authorization}
  158. dict_outputs = dict()
  159. response = tf_predict_pb2.PredictResponse()
  160. resp = requests.post(url, data=request_data, headers=headers)
  161. if resp.status_code != 200:
  162. print(resp.status_code,resp.content)
  163. log("调用pai-eas接口出错,authorization:"+str(authorization))
  164. return None
  165. else:
  166. response = tf_predict_pb2.PredictResponse()
  167. response.ParseFromString(resp.content)
  168. for _output in list_outputs:
  169. dict_outputs[_output] = get_values(response, _output)
  170. return dict_outputs
  171. def encodeInput(data,word_len,word_flag=True,userFool=False):
  172. result = []
  173. out_index = 0
  174. for item in data:
  175. if out_index in [0]:
  176. list_word = item[-word_len:]
  177. else:
  178. list_word = item[:word_len]
  179. temp = []
  180. if word_flag:
  181. for word in list_word:
  182. if userFool:
  183. temp.append(getIndexOfWord_fool(word))
  184. else:
  185. temp.append(getIndexOfWord(word))
  186. list_append = []
  187. temp_len = len(temp)
  188. while(temp_len<word_len):
  189. if userFool:
  190. list_append.append(0)
  191. else:
  192. list_append.append(getIndexOfWord("<pad>"))
  193. temp_len += 1
  194. if out_index in [0]:
  195. temp = list_append+temp
  196. else:
  197. temp = temp+list_append
  198. else:
  199. for words in list_word:
  200. temp.append(getIndexOfWords(words))
  201. list_append = []
  202. temp_len = len(temp)
  203. while(temp_len<word_len):
  204. list_append.append(getIndexOfWords("<pad>"))
  205. temp_len += 1
  206. if out_index in [0,1]:
  207. temp = list_append+temp
  208. else:
  209. temp = temp+list_append
  210. result.append(temp)
  211. out_index += 1
  212. return result
  213. def encodeInput_form(input,MAX_LEN=30):
  214. x = np.zeros([MAX_LEN])
  215. for i in range(len(input)):
  216. if i>=MAX_LEN:
  217. break
  218. x[i] = getIndexOfWord(input[i])
  219. return x
  220. def getVocabAndMatrix(model,Embedding_size = 60):
  221. '''
  222. @summary:获取子向量的词典和子向量矩阵
  223. '''
  224. vocab = ["<pad>"]+model.index2word
  225. embedding_matrix = np.zeros((len(vocab),Embedding_size))
  226. for i in range(1,len(vocab)):
  227. embedding_matrix[i] = model[vocab[i]]
  228. return vocab,embedding_matrix
  229. def getIndexOfWord(word):
  230. global vocab_word,file_vocab_word
  231. if vocab_word is None:
  232. if os.path.exists(file_vocab_word):
  233. vocab = load(file_vocab_word)
  234. vocab_word = dict((w, i) for i, w in enumerate(np.array(vocab)))
  235. else:
  236. model = getModel_word()
  237. vocab,_ = getVocabAndMatrix(model, Embedding_size=60)
  238. vocab_word = dict((w, i) for i, w in enumerate(np.array(vocab)))
  239. save(vocab,file_vocab_word)
  240. if word in vocab_word.keys():
  241. return vocab_word[word]
  242. else:
  243. return vocab_word['<pad>']
  244. def changeIndexFromWordToWords(tokens,word_index):
  245. '''
  246. @summary:转换某个字的字偏移为词偏移
  247. '''
  248. before_index = 0
  249. after_index = 0
  250. for i in range(len(tokens)):
  251. after_index = after_index+len(tokens[i])
  252. if before_index<=word_index and after_index>word_index:
  253. return i
  254. before_index = after_index
  255. return i+1
  256. def getIndexOfWords(words):
  257. global vocab_words,file_vocab_words
  258. if vocab_words is None:
  259. if os.path.exists(file_vocab_words):
  260. vocab = load(file_vocab_words)
  261. vocab_words = dict((w, i) for i, w in enumerate(np.array(vocab)))
  262. else:
  263. model = getModel_w2v()
  264. vocab,_ = getVocabAndMatrix(model, Embedding_size=128)
  265. vocab_words = dict((w, i) for i, w in enumerate(np.array(vocab)))
  266. save(vocab,file_vocab_words)
  267. if words in vocab_words.keys():
  268. return vocab_words[words]
  269. else:
  270. return vocab_words["<pad>"]
  271. def log(msg):
  272. '''
  273. @summary:打印信息
  274. '''
  275. logger.info(msg)
  276. def debug(msg):
  277. '''
  278. @summary:打印信息
  279. '''
  280. logger.debug(msg)
  281. def save(object_to_save, path):
  282. '''
  283. 保存对象
  284. @Arugs:
  285. object_to_save: 需要保存的对象
  286. @Return:
  287. 保存的路径
  288. '''
  289. with open(path, 'wb') as f:
  290. pickle.dump(object_to_save, f)
  291. def load(path):
  292. '''
  293. 读取对象
  294. @Arugs:
  295. path: 读取的路径
  296. @Return:
  297. 读取的对象
  298. '''
  299. with open(path, 'rb') as f:
  300. object1 = pickle.load(f)
  301. return object1
  302. fool_char_to_id = load(os.path.dirname(__file__)+"/fool_char_to_id.pk")
  303. def getIndexOfWord_fool(word):
  304. if word in fool_char_to_id.keys():
  305. return fool_char_to_id[word]
  306. else:
  307. return fool_char_to_id["[UNK]"]
  308. def find_index(list_tofind,text):
  309. '''
  310. @summary: 查找所有词汇在字符串中第一次出现的位置
  311. @param:
  312. list_tofind:待查找词汇
  313. text:字符串
  314. @return: list,每个词汇第一次出现的位置
  315. '''
  316. result = []
  317. for item in list_tofind:
  318. index = text.find(item)
  319. if index>=0:
  320. result.append(index)
  321. else:
  322. result.append(-1)
  323. return result
  324. def combine(list1,list2):
  325. '''
  326. @summary:将两个list中的字符串两两拼接
  327. @param:
  328. list1:字符串list
  329. list2:字符串list
  330. @return:拼接结果list
  331. '''
  332. result = []
  333. for item1 in list1:
  334. for item2 in list2:
  335. result.append(str(item1)+str(item2))
  336. return result
  337. def getDigitsDic(unit):
  338. '''
  339. @summary:拿到中文对应的数字
  340. '''
  341. DigitsDic = {"零":0, "壹":1, "贰":2, "叁":3, "肆":4, "伍":5, "陆":6, "柒":7, "捌":8, "玖":9,
  342. "〇":0, "一":1, "二":2, "三":3, "四":4, "五":5, "六":6, "七":7, "八":8, "九":9}
  343. return DigitsDic.get(unit)
  344. def getMultipleFactor(unit):
  345. '''
  346. @summary:拿到单位对应的值
  347. '''
  348. MultipleFactor = {"兆":Decimal(1000000000000),"亿":Decimal(100000000),"万":Decimal(10000),"仟":Decimal(1000),"千":Decimal(1000),"佰":Decimal(100),"百":Decimal(100),"拾":Decimal(10),"十":Decimal(10),"元":Decimal(1),"圆":Decimal(1),"角":round(Decimal(0.1),1),"分":round(Decimal(0.01),2)}
  349. return MultipleFactor.get(unit)
  350. def getUnifyMoney(money):
  351. '''
  352. @summary:将中文金额字符串转换为数字金额
  353. @param:
  354. money:中文金额字符串
  355. @return: decimal,数据金额
  356. '''
  357. MAX_MONEY = 1000000000000
  358. MAX_NUM = 12
  359. #去掉逗号
  360. money = re.sub("[,,]","",money)
  361. money = re.sub("[^0-9.零壹贰叁肆伍陆柒捌玖拾佰仟萬億圆十百千万亿元角分]","",money)
  362. result = Decimal(0)
  363. chnDigits = ["零", "壹", "贰", "叁", "肆", "伍", "陆", "柒", "捌", "玖"]
  364. # chnFactorUnits = ["兆", "亿", "万", "仟", "佰", "拾","圆","元","角","分"]
  365. chnFactorUnits = ["兆", "亿", "万", "仟", '千', "佰", '百', "拾", '十',"圆", "元", "角", "分"] # 20240611 修复大写提取错误 '陆拾陆亿伍千柒佰零叁万肆千叁佰陆拾伍元' Decimal('11607430365')
  366. LowMoneypattern = re.compile("^[\d,]+(\.\d+)?$")
  367. BigMoneypattern = re.compile("^零?(?P<BigMoney>[%s])$"%("".join(chnDigits)))
  368. try:
  369. if re.search(LowMoneypattern,money) is not None:
  370. return Decimal(money)
  371. elif re.search(BigMoneypattern,money) is not None:
  372. return getDigitsDic(re.search(BigMoneypattern,money).group("BigMoney"))
  373. for factorUnit in chnFactorUnits:
  374. if re.search(re.compile(".*%s.*"%(factorUnit)),money) is not None:
  375. subMoneys = re.split(re.compile("%s(?!.*%s.*)"%(factorUnit,factorUnit)),money)
  376. if re.search(re.compile("^(\d+)(\.\d+)?$"),subMoneys[0]) is not None:
  377. if MAX_MONEY/getMultipleFactor(factorUnit)<Decimal(subMoneys[0]):
  378. return Decimal(0)
  379. result += Decimal(subMoneys[0])*(getMultipleFactor(factorUnit))
  380. elif len(subMoneys[0])==1:
  381. if re.search(re.compile("^[%s]$"%("".join(chnDigits))),subMoneys[0]) is not None:
  382. result += Decimal(getDigitsDic(subMoneys[0]))*(getMultipleFactor(factorUnit))
  383. # subMoneys[0]中无金额单位,不可再拆分
  384. elif subMoneys[0]=="":
  385. result += 0
  386. elif re.search(re.compile("[%s]"%("".join(chnFactorUnits))),subMoneys[0]) is None:
  387. # print(subMoneys)
  388. # subMoneys[0] = subMoneys[0][0]
  389. result += Decimal(getUnifyMoney(subMoneys[0])) * (getMultipleFactor(factorUnit))
  390. else:
  391. result += Decimal(getUnifyMoney(subMoneys[0]))*(getMultipleFactor(factorUnit))
  392. if len(subMoneys)>1:
  393. if re.search(re.compile("^(\d+(,)?)+(\.\d+)?[百千万亿]?\s?(元)?$"),subMoneys[1]) is not None:
  394. result += Decimal(subMoneys[1])
  395. elif len(subMoneys[1])==1:
  396. if re.search(re.compile("^[%s]$"%("".join(chnDigits))),subMoneys[1]) is not None:
  397. result += Decimal(getDigitsDic(subMoneys[1]))
  398. else:
  399. result += Decimal(getUnifyMoney(subMoneys[1]))
  400. break
  401. except Exception as e:
  402. # traceback.print_exc()
  403. return Decimal(0)
  404. return result
  405. def getModel_w2v():
  406. '''
  407. @summary:加载词向量
  408. '''
  409. global model_w2v,lock_model_w2v
  410. with lock_model_w2v:
  411. if model_w2v is None:
  412. model_w2v = gensim.models.KeyedVectors.load_word2vec_format(getw2vfilepath(),binary=True)
  413. return model_w2v
  414. def getModel_word():
  415. '''
  416. @summary:加载字向量
  417. '''
  418. global model_word,lock_model_w2v
  419. with lock_model_word:
  420. if model_word is None:
  421. model_word = gensim.models.KeyedVectors.load_word2vec_format(model_word_file,binary=True)
  422. return model_word
  423. # getModel_w2v()
  424. # getModel_word()
  425. def findAllIndex(substr,wholestr):
  426. '''
  427. @summary: 找到字符串的子串的所有begin_index
  428. @param:
  429. substr:子字符串
  430. wholestr:子串所在完整字符串
  431. @return: list,字符串的子串的所有begin_index
  432. '''
  433. copystr = wholestr
  434. result = []
  435. indexappend = 0
  436. while(True):
  437. index = copystr.find(substr)
  438. if index<0:
  439. break
  440. else:
  441. result.append(indexappend+index)
  442. indexappend += index+len(substr)
  443. copystr = copystr[index+len(substr):]
  444. return result
  445. def spanWindow(tokens,begin_index,end_index,size,center_include=False,word_flag = False,use_text = False,text = None):
  446. '''
  447. @summary:取得某个实体的上下文词汇
  448. @param:
  449. tokens:句子分词list
  450. begin_index:实体的开始index
  451. end_index:实体的结束index
  452. size:左右两边各取多少个词
  453. center_include:是否包含实体
  454. word_flag:词/字,默认是词
  455. @return: list,实体的上下文词汇
  456. '''
  457. if use_text:
  458. assert text is not None
  459. length_tokens = len(tokens)
  460. if begin_index>size:
  461. begin = begin_index-size
  462. else:
  463. begin = 0
  464. if end_index+size<length_tokens:
  465. end = end_index+size+1
  466. else:
  467. end = length_tokens
  468. result = []
  469. if not word_flag:
  470. result.append(tokens[begin:begin_index])
  471. if center_include:
  472. if use_text:
  473. result.append(text)
  474. else:
  475. result.append(tokens[begin_index:end_index+1])
  476. result.append(tokens[end_index+1:end])
  477. else:
  478. result.append("".join(tokens[begin:begin_index]))
  479. if center_include:
  480. if use_text:
  481. result.append(text)
  482. else:
  483. result.append("".join(tokens[begin_index:end_index+1]))
  484. result.append("".join(tokens[end_index+1:end]))
  485. #print(result)
  486. return result
  487. def get_context(sentence_text, begin_index, end_index, size=20, center_include=False):
  488. '''
  489. 返回实体上下文信息
  490. :param sentence_text: 句子文本
  491. :param begin_index: 实体字开始位置
  492. :param end_index: 实体字结束位置
  493. :param size: 字偏移量
  494. :param center_include:
  495. :return:
  496. '''
  497. result = []
  498. begin = begin_index - size if begin_index>size else 0
  499. end = end_index + size
  500. result.append(sentence_text[begin: begin_index])
  501. if center_include:
  502. result.append(sentence_text[begin_index: end_index])
  503. result.append(sentence_text[end_index: end])
  504. return result
  505. #根据规则补全编号或名称两边的符号
  506. def fitDataByRule(data):
  507. symbol_dict = {"(":")",
  508. "(":")",
  509. "[":"]",
  510. "【":"】",
  511. ")":"(",
  512. ")":"(",
  513. "]":"[",
  514. "】":"【"}
  515. leftSymbol_pattern = re.compile("[\((\[【]")
  516. rightSymbol_pattern = re.compile("[\))\]】]")
  517. leftfinds = re.findall(leftSymbol_pattern,data)
  518. rightfinds = re.findall(rightSymbol_pattern,data)
  519. result = data
  520. if len(leftfinds)+len(rightfinds)==0:
  521. return data
  522. elif len(leftfinds)==len(rightfinds):
  523. return data
  524. elif abs(len(leftfinds)-len(rightfinds))==1:
  525. if len(leftfinds)>len(rightfinds):
  526. if symbol_dict.get(data[0]) is not None:
  527. result = data[1:]
  528. else:
  529. #print(symbol_dict.get(leftfinds[0]))
  530. result = data+symbol_dict.get(leftfinds[0])
  531. else:
  532. if symbol_dict.get(data[-1]) is not None:
  533. result = data[:-1]
  534. else:
  535. result = symbol_dict.get(rightfinds[0])+data
  536. result = re.sub("[。]","",result)
  537. return result
  538. from datetime import date
  539. # 时间合法性判断
  540. def isValidDate(year, month, day):
  541. try:
  542. date(year, month, day)
  543. except:
  544. return False
  545. else:
  546. return True
  547. time_format_pattern = re.compile("((?P<year>20\d{2}|\d{2}|二[零〇0][零〇一二三四五六七八九0]{2})\s*[-/年.]\s*(?P<month>\d{1,2}|[一二三四五六七八九十]{1,3})\s*[-/月.]?\s*(?P<day>\d{1,2}|[一二三四五六七八九十]{1,3})?)")
  548. from BiddingKG.dl.ratio.re_ratio import getUnifyNum
  549. import calendar
  550. def get_maxday(year, month):
  551. # calendar.monthrange(year, month) 返回一个元组,其中第一个元素是那个月第一天的星期几(0-6代表周一到周日),
  552. # 第二个元素是那个月的天数。
  553. _, last_day = calendar.monthrange(year, month)
  554. return last_day
  555. def timeFormat(_time, default_first_day=True):
  556. '''
  557. 日期格式化:年-月-日
  558. :param _time:
  559. :param default_first_day: True取当月第一天,否则取最后一天
  560. :return:
  561. '''
  562. current_year = time.strftime("%Y",time.localtime())
  563. all_match = re.finditer(time_format_pattern,_time)
  564. for _match in all_match:
  565. if len(_match.group())>0:
  566. legal = True
  567. year = ""
  568. month = ""
  569. day = ""
  570. for k,v in _match.groupdict().items():
  571. if k=="year":
  572. year = v
  573. if k=="month":
  574. month = v
  575. if k=="day":
  576. day = v
  577. if year!="":
  578. if re.search("^\d+$",year):
  579. if len(year)==2:
  580. year = "20"+year
  581. if int(year)-int(current_year)>10:
  582. legal = False
  583. else:
  584. _year = ""
  585. for word in year:
  586. if word == '0':
  587. _year += word
  588. else:
  589. _year += str(getDigitsDic(word))
  590. year = _year
  591. else:
  592. legal = False
  593. if month!="":
  594. if re.search("^\d+$", month):
  595. if int(month)>12:
  596. legal = False
  597. else:
  598. month = int(getUnifyNum(month))
  599. if month>=1 and month<=12:
  600. month = str(month)
  601. else:
  602. legal = False
  603. else:
  604. legal = False
  605. if day == None:
  606. day = "01" if (default_first_day or legal == False) else str(get_maxday(int(year), int(month)))
  607. if day!="":
  608. if re.search("^\d+$", day):
  609. if int(day)>31:
  610. legal = False
  611. else:
  612. day = int(getUnifyNum(day))
  613. if day >= 1 and day <= 31:
  614. day = str(day)
  615. else:
  616. legal = False
  617. else:
  618. legal = False
  619. # print(year,month,day)
  620. if not isValidDate(int(year),int(month),int(day)):
  621. legal = False
  622. if legal:
  623. return "%s-%s-%s"%(year,month.rjust(2,"0"),day.rjust(2,"0"))
  624. return ""
  625. def embedding(datas,shape):
  626. '''
  627. @summary:查找词汇对应的词向量
  628. @param:
  629. datas:词汇的list
  630. shape:结果的shape
  631. @return: array,返回对应shape的词嵌入
  632. '''
  633. model_w2v = getModel_w2v()
  634. embed = np.zeros(shape)
  635. length = shape[1]
  636. out_index = 0
  637. #print(datas)
  638. for data in datas:
  639. index = 0
  640. for item in data:
  641. item_not_space = re.sub("\s*","",item)
  642. if index>=length:
  643. break
  644. if item_not_space in model_w2v.vocab:
  645. embed[out_index][index] = model_w2v[item_not_space]
  646. index += 1
  647. else:
  648. #embed[out_index][index] = model_w2v['unk']
  649. index += 1
  650. out_index += 1
  651. return embed
  652. def embedding_word(datas,shape):
  653. '''
  654. @summary:查找词汇对应的词向量
  655. @param:
  656. datas:词汇的list
  657. shape:结果的shape
  658. @return: array,返回对应shape的词嵌入
  659. '''
  660. model_w2v = getModel_word()
  661. embed = np.zeros(shape)
  662. length = shape[1]
  663. out_index = 0
  664. #print(datas)
  665. for data in datas:
  666. index = 0
  667. for item in str(data)[-shape[1]:]:
  668. if index>=length:
  669. break
  670. if item in model_w2v.vocab:
  671. embed[out_index][index] = model_w2v[item]
  672. index += 1
  673. else:
  674. # embed[out_index][index] = model_w2v['unk']
  675. index += 1
  676. out_index += 1
  677. return embed
  678. def embedding_word_forward(datas,shape):
  679. '''
  680. @summary:查找词汇对应的词向量
  681. @param:
  682. datas:词汇的list
  683. shape:结果的shape
  684. @return: array,返回对应shape的词嵌入
  685. '''
  686. model_w2v = getModel_word()
  687. embed = np.zeros(shape)
  688. length = shape[1]
  689. out_index = 0
  690. #print(datas)
  691. for data in datas:
  692. index = 0
  693. for item in str(data)[:shape[1]]:
  694. if index>=length:
  695. break
  696. if item in model_w2v.vocab:
  697. embed[out_index][index] = model_w2v[item]
  698. index += 1
  699. else:
  700. # embed[out_index][index] = model_w2v['unk']
  701. index += 1
  702. out_index += 1
  703. return embed
  704. def formEncoding(text,shape=(100,60),expand=False):
  705. embedding = np.zeros(shape)
  706. word_model = getModel_word()
  707. for i in range(len(text)):
  708. if i>=shape[0]:
  709. break
  710. if text[i] in word_model.vocab:
  711. embedding[i] = word_model[text[i]]
  712. if expand:
  713. embedding = np.expand_dims(embedding,0)
  714. return embedding
  715. def partMoney(entity_text,input2_shape = [7]):
  716. '''
  717. @summary:对金额分段
  718. @param:
  719. entity_text:数值金额
  720. input2_shape:分类数
  721. @return: array,分段之后的独热编码
  722. '''
  723. money = float(entity_text)
  724. parts = np.zeros(input2_shape)
  725. if money<100:
  726. parts[0] = 1
  727. elif money<1000:
  728. parts[1] = 1
  729. elif money<10000:
  730. parts[2] = 1
  731. elif money<100000:
  732. parts[3] = 1
  733. elif money<1000000:
  734. parts[4] = 1
  735. elif money<10000000:
  736. parts[5] = 1
  737. else:
  738. parts[6] = 1
  739. return parts
  740. def uniform_num(num):
  741. d1 = {'一': '1', '二': '2', '三': '3', '四': '4', '五': '5', '六': '6', '七': '7', '八': '8', '九': '9', '十': '10'}
  742. # d2 = {'A': '1', 'B': '2', 'C': '3', 'D': '4', 'E': '5', 'F': '6', 'G': '7', 'H': '8', 'I': '9', 'J': '10'}
  743. d3 = {'Ⅰ': '1', 'Ⅱ': '2', 'Ⅲ': '3', 'Ⅳ': '4', 'Ⅴ': '5', 'Ⅵ': '6', 'Ⅶ': '7'}
  744. if num.isdigit():
  745. if re.search('^0[\d]$', num):
  746. num = num[1:]
  747. return num
  748. elif re.search('^[一二三四五六七八九十]+$', num):
  749. _digit = re.search('^[一二三四五六七八九十]+$', num).group(0)
  750. if len(_digit) == 1:
  751. num = d1[_digit]
  752. elif len(_digit) == 2 and _digit[0] == '十':
  753. num = '1'+ d1[_digit[1]]
  754. elif len(_digit) == 2 and _digit[1] == '十':
  755. num = d1[_digit[0]] + '0'
  756. elif len(_digit) == 3 and _digit[1] == '十':
  757. num = d1[_digit[0]] + d1[_digit[2]]
  758. elif re.search('[ⅠⅡⅢⅣⅤⅥⅦ]', num):
  759. num = re.search('[ⅠⅡⅢⅣⅤⅥⅦ]', num).group(0)
  760. num = d3[num]
  761. return num
  762. def uniform_package_name(package_name):
  763. '''
  764. 统一规范化包号。数值类型统一为阿拉伯数字,字母统一为大写,包含施工监理等抽到前面, 例 A包监理一标段 统一为 监理A1 ; 包Ⅱ 统一为 2
  765. :param package_name: 字符串类型 包号
  766. :return:
  767. '''
  768. package_name_raw = package_name
  769. package_name = re.sub('pdf|doc|docs|xlsx|rar|\d{4}年', ' ', package_name)
  770. package_name = package_name.replace('标段(包)', '标段').replace('№', '')
  771. package_name = re.sub('\[|【', '', package_name)
  772. kw = re.search('(施工|监理|监测|勘察|设计|劳务)', package_name)
  773. name = ""
  774. if kw:
  775. name += kw.group(0)
  776. if re.search('^[a-zA-Z0-9-]{5,}$', package_name): # 五个字符以上编号
  777. _digit = re.search('^[a-zA-Z0-9-]{5,}$', package_name).group(0).upper()
  778. # print('规范化包号1', _digit)
  779. name += _digit
  780. elif re.search('(?P<eng>[a-zA-Z])包[:)]?第?(?P<num>([0-9]{1,4}|[一二三四五六七八九十]{1,4}|[ⅠⅡⅢⅣⅤⅥⅦ]{1,4}))标段?', package_name): # 处理类似 A包2标段
  781. ser = re.search('(?P<eng>[a-zA-Z])包[:)]?第?(?P<num>([0-9]{1,4}|[一二三四五六七八九十]{1,4}|[ⅠⅡⅢⅣⅤⅥⅦ]{1,4}))标段?', package_name)
  782. # print('规范化包号2', ser.group(0))
  783. _char = ser.groupdict().get('eng')
  784. if _char:
  785. _char = _char.upper()
  786. _digit = ser.groupdict().get('num')
  787. _digit = uniform_num(_digit)
  788. name += _char.upper() + _digit
  789. elif re.search('第?(?P<eng>[0-9a-zA-Z-]{1,4})?(?P<num>([0-9]{1,4}|[一二三四五六七八九十]{1,4}|[ⅠⅡⅢⅣⅤⅥⅦ]{1,4}))(标[段号的包项]?|合同[包段]|([分子]?[包标]))', package_name): # 处理类似 A包2标段
  790. ser = re.search('第?(?P<eng>[0-9a-zA-Z-]{1,4})?(?P<num>([0-9]{1,4}|[一二三四五六七八九十]{1,4}|[ⅠⅡⅢⅣⅤⅥⅦ]{1,4}))(标[段号的包项]?|合同[包段]|([分子]?[包标]))', package_name)
  791. # print('规范化包号3', ser.group(0))
  792. _char = ser.groupdict().get('eng')
  793. if _char:
  794. _char = _char.upper()
  795. _digit = ser.groupdict().get('num')
  796. _digit = uniform_num(_digit)
  797. if _char:
  798. name += _char.upper()
  799. name += _digit
  800. elif re.search('(标[段号的包项]?|项目|子项目?|([分子]?包|包[组件号]))编?号?[::]?(?P<eng>[0-9a-zA-Z-]{1,4})?(?P<num>([0-9]{1,4}|[一二三四五六七八九十]{1,4}|[ⅠⅡⅢⅣⅤⅥⅦ]{1,4}))', package_name): # 数字的统一的阿拉伯数字
  801. ser = re.search('(标[段号的包项]?|项目|子项目?|([分子]?包|包[组件号]))编?号?[::]?(?P<eng>[0-9a-zA-Z-]{1,4})?(?P<num>([0-9]{1,4}|[一二三四五六七八九十]{1,4}|[ⅠⅡⅢⅣⅤⅥⅦ]{1,4}))',package_name)
  802. # print('规范化包号4', ser.group(0))
  803. _char = ser.groupdict().get('eng')
  804. if _char:
  805. _char = _char.upper()
  806. _digit = ser.groupdict().get('num')
  807. _digit = uniform_num(_digit)
  808. if _char:
  809. name += _char.upper()
  810. name += _digit
  811. elif re.search('(标[段号的包项]|([分子]?包|包[组件号]))编?号?[::]?(?P<eng>[a-zA-Z-]{1,5})', package_name): # 数字的统一的阿拉伯数字
  812. _digit = re.search('(标[段号的包项]|([分子]?包|包[组件号]))编?号?[::]?(?P<eng>[a-zA-Z-]{1,5})', package_name).group('eng').upper()
  813. # print('规范化包号5', _digit)
  814. name += _digit
  815. elif re.search('(?P<eng>[a-zA-Z]{1,4})(标[段号的包项]|([分子]?[包标]|包[组件号]))', package_name): # 数字的统一的阿拉伯数字
  816. _digit = re.search('(?P<eng>[a-zA-Z]{1,4})(标[段号的包项]|([分子]?[包标]|包[组件号]))', package_name).group('eng').upper()
  817. # print('规范化包号6', _digit)
  818. name += _digit
  819. elif re.search('^([0-9]{1,4}|[一二三四五六七八九十]{1,4}|[ⅠⅡⅢⅣⅤⅥⅦ]{1,4})$', package_name): # 数字的统一的阿拉伯数字
  820. _digit = re.search('^([0-9]{1,4}|[一二三四五六七八九十]{1,4}|[ⅠⅡⅢⅣⅤⅥⅦ]{1,4})$', package_name).group(0)
  821. # print('规范化包号7', _digit)
  822. _digit = uniform_num(_digit)
  823. name += _digit
  824. elif re.search('^[a-zA-Z0-9-]+$', package_name):
  825. _char = re.search('^[a-zA-Z0-9-]+$', package_name).group(0)
  826. # print('规范化包号8', _char)
  827. name += _char.upper()
  828. if name == "":
  829. return package_name_raw
  830. else:
  831. if name.isdigit():
  832. name = str(int(name))
  833. # print('原始包号:%s, 处理后:%s'%(package_name, name))
  834. return name
  835. def money_process(money_text, header):
  836. '''
  837. 输入金额文本及金额列表头,返回统一数字化金额及金额单位
  838. :param money_text:金额字符串
  839. :param header:金额列表头,用于提取单位
  840. :return:
  841. '''
  842. money = 0
  843. money_unit = ""
  844. moneys, _ = get_money_entity('%s:%s' % (header, money_text))
  845. if len(moneys) == 1:
  846. money = float(moneys[0][0])
  847. money_unit = moneys[0][3]
  848. elif len(moneys) == 2 and moneys[0][0]==moneys[1][0]:
  849. money = float(moneys[0][0])
  850. money_unit = moneys[0][3]
  851. # # re_price = re.search("[零壹贰叁肆伍陆柒捌玖拾佰仟萬億圆十百千万亿元角分]{3,}|\d[\d,]*(?:\.\d+)?[((]?万?", money_text)
  852. # money_text = re.sub('\s', '', money_text) # 2024/04/19 修复 457699044 556.46751 万元 金额与单位有空格造成万漏提取
  853. # if re.search('[零壹贰叁肆伍陆柒捌玖拾佰仟萬億圆十百千万亿元角分]{3,}', money_text) and re.search('\d{1,3}([,,]\d{3})+(\.\d+)?|\d+(\.\d+)?[((]?万?', money_text):
  854. # money_text = re.search('[零壹贰叁肆伍陆柒捌玖拾佰仟萬億圆十百千万亿元角分]{3,}', money_text).group(0) # 如果表格同时包含大小写金额,取大写金额,避免单位取错 463310590 790000(柒拾玖万元整)
  855. # re_price = re.search("[零壹贰叁肆伍陆柒捌玖拾佰仟萬億圆十百千万亿元角分]{3,}|\d{1,3}([,,]\d{3})+(\.\d+)?|\d+(\.\d+)?[((]?万?", money_text)
  856. # if re_price:
  857. # money_re = re_price.group(0)
  858. # if (re.search('万元|[((]万[))]', header) or re.search('万元|[((]万[))]', money_text)) and '万' not in money_re: # 修复37797825 控制价(万) # 修复 460307391 万元不在表头,在数字前面
  859. # money_re += '万元'
  860. # elif (re.search('亿元|[((]亿[))]', header) or re.search('亿元|[((]亿[))]', money_text)) and '亿' not in money_re: # 修复37797825 控制价(万) # 修复 460307391 万元不在表头,在数字前面
  861. # money_re += '亿元'
  862. # # money = float(getUnifyMoney(money_text))
  863. # money = float(getUnifyMoney(money_re))
  864. # if money > 10000000000000: # 大于万亿的去除
  865. # money = 0
  866. # # money_unit = '万元' if '万' in money_re and re.search('[零壹贰叁肆伍陆柒捌玖拾佰仟萬億圆十百千万亿元角分]{3,}', money_text)==None else '元'
  867. # if re.search('[零壹贰叁肆伍陆柒捌玖拾佰仟萬億圆十百千万亿元角分]{3,}', money_text)==None:
  868. # if '万' in money_re:
  869. # money_unit = '万元'
  870. # elif '亿' in money_re:
  871. # money_unit = '亿元'
  872. # else:
  873. # money_unit = '元'
  874. return (money, money_unit)
  875. package_number_pattern = re.compile(
  876. '((施工|监理|监测|勘察|设计|劳务)(标段)?:?第?([一二三四五六七八九十]{1,3}|[ⅠⅡⅢⅣⅤⅥⅦ]{1,3}|[a-zA-Z0-9]{1,9}\-?[a-zA-Z0-9-]{,9})?[分子]?(标[段包项]?|包[组件标]?|合同[包段]))\
  877. |(([a-zA-Z]包[:()]?)?第?([一二三四五六七八九十]{1,3}|[ⅠⅡⅢⅣⅤⅥⅦ]{1,3}|[a-zA-Z0-9]{1,9}\-?[a-zA-Z0-9-]{,9})[分子]?(标[段包项]?|合同[包段]))\
  878. |(([,;。、:(]|第)?([一二三四五六七八九十]{1,3}|[ⅠⅡⅢⅣⅤⅥⅦ]{1,3}|[a-zA-Z0-9]{1,9}\-?[a-zA-Z0-9-]{,9})[分子]?(标[段包项]?|包[组件标]?|合同[包段]))\
  879. |((标[段包项]|品目|标段(包)|包[组件标]|[标分子(]包)(\[|【)?:?([一二三四五六七八九十]{1,3}|[ⅠⅡⅢⅣⅤⅥⅦ]{1,3}|[a-zA-Z0-9]{1,9}\-?[a-zA-Z0-9-]{,9}))\
  880. |([,;。、:(]|^)(标的?|项目|子项目?)(\[|【)?:?([一二三四五六七八九十]+|[0-9]{1,9})\
  881. |((([标分子(]|合同|项目|采购)包|[,。]标的|子项目|[分子]标|标[段包项]|包[组件标]?)编?号[::]?[a-zA-Z0-9一二三四五六七八九十ⅠⅡⅢⅣⅤⅥⅦ]{1,9}[a-zA-Z0-9一二三四五六七八九十ⅠⅡⅢⅣⅤⅥⅦ-]{0,9})\
  882. |[,;。、:(]?(合同|分|子)?包:?([一二三四五六七八九十]{1,3}|[ⅠⅡⅢⅣⅤⅥⅦ]{1,3}|[a-zA-Z0-9]{1,9}\-?[a-zA-Z0-9-]{,9})')
  883. filter_package_pattern = 'CA标|(每个?|所有|相关|个|各|不分)[分子]?(标[段包项]?|包[组件标]?|合同包)|(质量|责任)三包|包[/每]|标段(划分|范围)|(承|压缩|软|皮|书|挂)包\
  884. |标[识注签贴配]|[商油]标号|第X包|第[一二三四五六七八九十]+至[一二三四五六七八九十]+(标[段包项]?|包[组件标]?|合同[包段])\
  885. |\.(docx|doc|pdf|xlsx|xls|jpg)|[一二三四五]次|五金|\d+[年月]|[\d.,]+万?元|\d+\.\d+' # 过滤错误的非包号
  886. def find_package(content):
  887. '''
  888. 通过正则找包和标段号
  889. :param content:
  890. :return:
  891. '''
  892. packages = []
  893. content = content.replace('号,', '号:').replace(':', ':').replace('(', '(').replace(')', ')')
  894. # .replace('-包',' 包').replace('包-', '包 ').replace('-标', ' 标').replace('标段-', '标段 ').replace('-合同包', ' 合同包') # 72760191 标段:№10
  895. content = re.sub('[一二三四五六七八九十\d](标[段包项]|包[组件标])编号', ' 标段编号', content)
  896. for it in re.finditer(filter_package_pattern, content):
  897. content = content.replace(it.group(0), ' ' * len(it.group(0)))
  898. for iter in re.finditer(package_number_pattern, content):
  899. if re.search('(业绩|信誉要求):|业绩(如下)?\d*[、:]', content[:iter.start()]): # 前面有业绩或信誉的标段去掉
  900. continue
  901. # print('提取到标段:%s, 前后文:%s' % (iter.group(), content[iter.start() - 5:iter.end() + 5]))
  902. if re.match('\d', iter.group(0)) and re.search('\d\.$', content[:iter.start()]): # 排除2.10标段3 5.4标段划分 这种情况
  903. # print('过滤掉错误包:', iter.group())
  904. continue
  905. if re.search('[承每书/]包|XX|xx', iter.group(0)) or re.search('\d包[/每]\w|一包[0-9一二三四五六七八九十]+', content[
  906. iter.start():iter.end() + 3]) or re.search(
  907. '[a-zA-Z0-9一二三四五六七八九十ⅠⅡⅢⅣⅤⅥⅦ-]{6,}', iter.group(0)):
  908. # print('过滤掉错误包:', iter.group())
  909. continue
  910. elif iter.end() + 2 < len(content) and re.search('标的物|包装|划分|标(书|准|志|记|识|签|贴|帜|本|底|价|量)',
  911. content[iter.start():iter.end() + 2]):
  912. # print('过滤掉错误包:', iter.group())
  913. continue
  914. elif re.search('同一(标段?|包)', content[max(0, iter.start() - 2):iter.end()]): # 不得参加同一标段
  915. # print('过滤掉错误包:', iter.group())
  916. continue
  917. elif re.search('三包', content[max(0, iter.start() - 2):iter.end()]) and re.search('第三包', content[max(0,
  918. iter.start() - 2):iter.end()]) == None: # 规规章和“三包”规定
  919. # print('过滤掉错误包:', iter.group())
  920. continue
  921. elif re.search('[1-9]\d{2,}$|\d{4,}|^[1-9]\d{2,}|合同包[A-Za-z]{2,}', iter.group(0)):
  922. # print('过滤掉错误包号5:', iter.group(0))
  923. continue
  924. elif re.search('单位:包|1包\d|[张箱]', content[max(0, iter.start()-3): iter.end()+2]): # 处理 463166661 包号错误 钢丝,单位:包X10根。
  925. # print('过滤掉错误包号,单位:包|1包', iter.group(0))
  926. continue
  927. packages.append(iter)
  928. # print('提取到标段:%s, 前后文:%s' % (iter.group(), content[iter.start() - 5:iter.end() + 5]))
  929. return packages
  930. def cut_repeat_name(s):
  931. '''
  932. 公司连续重复名称去重
  933. :param s:
  934. :return:
  935. '''
  936. if len(s) >= 8:
  937. n = s.count(s[-4:])
  938. id = s.find(s[-4:]) + 4
  939. sub_s = s[:id]
  940. if n>=2 and s == sub_s * n:
  941. s = sub_s
  942. return s
  943. def del_tabel_achievement(soup):
  944. if re.search('中标|成交|入围|结果|评标|开标|候选人', soup.text[:800]) == None or re.search('业绩', soup.text)==None:
  945. return None
  946. p1 = '(中标|成交)(单位|候选人)的?(企业|项目|项目负责人|\w{,5})?业绩|类似(项目)?业绩|\w{,10}业绩$|业绩(公示|情况|荣誉)'
  947. '''删除前面标签 命中业绩规则;当前标签为表格且公布业绩相关信息的去除'''
  948. for tag in soup.find_all('table'):
  949. pre_text = ""
  950. if tag.findPreviousSibling() != None:
  951. pre_text = tag.findPreviousSibling().text.strip()
  952. if pre_text == "" and tag.findPreviousSibling().findPreviousSibling() != None: # 修复表格前一标签没内容,再前一个才有内容情况
  953. pre_text = tag.findPreviousSibling().findPreviousSibling().text.strip()
  954. tr_text = tag.find('tr').text.strip() if tag.find('tr') != None else ""
  955. # print(re.search(p1, pre_text),pre_text, len(pre_text), re.findall('序号|中标候选人名称|项目名称|工程名称|合同金额|建设单位|业主', tr_text))
  956. if re.search(p1, pre_text) and len(pre_text) < 20 and tag.find('tr') != None and len(tr_text)<100:
  957. _count = 0
  958. for td in tag.find('tr').find_all('td'):
  959. td_text = td.text.strip()
  960. if len(td_text) > 25:
  961. break
  962. if len(td_text) < 25 and re.search('中标候选人|第[一二三四五1-5]候选人|(项目|业绩|工程)名称|\w{,10}业绩$|合同金额|建设单位|采购单位|业主|甲方', td_text):
  963. _count += 1
  964. if _count >=2:
  965. pre_tag = tag.findPreviousSibling().extract()
  966. del_tag = tag.extract()
  967. # print('删除表格业绩内容', pre_tag.text + del_tag.text)
  968. break
  969. elif re.search('业绩名称', tr_text) and re.search('建设单位|采购单位|业主', tr_text) and len(tr_text)<100:
  970. del_tag = tag.extract()
  971. # print('删除表格业绩内容', del_tag.text)
  972. del_trs = []
  973. '''删除表格某些行公布的业绩信息'''
  974. for tag in soup.find_all('table'):
  975. text = tag.text
  976. if re.search('业绩', text) == None:
  977. continue
  978. # for tr in tag.find_all('tr'):
  979. trs = tag.find_all('tr')
  980. i = 0
  981. while i < len(trs):
  982. tr = trs[i]
  983. if len(tr.find_all('td'))==2 and tr.td!=None and tr.td.findNextSibling()!=None:
  984. td1_text =tr.td.text
  985. td2_text =tr.td.findNextSibling().text
  986. if re.search('业绩', td1_text)!=None and len(td1_text)<10 and len(re.findall('(\d、|(\d))?[-\w()、]+(工程|项目|勘察|设计|施工|监理|总承包|采购|更新)', td2_text))>=2:
  987. # del_tag = tr.extract()
  988. # print('删除表格业绩内容', del_tag.text)
  989. del_trs.append(tr)
  990. elif tr.td != None and re.search('^业绩|业绩$', tr.td.text.strip()) and len(tr.td.text.strip())<25:
  991. rows = tr.td.attrs.get('rowspan', '')
  992. cols = tr.td.attrs.get('colspan', '')
  993. if rows.isdigit() and int(rows)>2:
  994. for j in range(int(rows)):
  995. if i+j < len(trs):
  996. del_trs.append(trs[i+j])
  997. i += j
  998. elif cols.isdigit() and int(cols)>3 and len(tr.find_all('td'))==1 and i+2 < len(trs):
  999. next_tr_cols = 0
  1000. td_num = 0
  1001. for td in trs[i+1].find_all('td'):
  1002. td_num += 1
  1003. if td.attrs.get('colspan', '').isdigit():
  1004. next_tr_cols += int(td.attrs.get('colspan', ''))
  1005. if next_tr_cols == int(cols):
  1006. del_trs.append(tr)
  1007. for j in range(1,len(trs)-i):
  1008. if len(trs[i+j].find_all('td')) == 1:
  1009. break
  1010. elif len(trs[i+j].find_all('td')) >= td_num-1:
  1011. del_trs.append(trs[i+j])
  1012. else:
  1013. break
  1014. i += j
  1015. i += 1
  1016. for tr in del_trs:
  1017. del_tag = tr.extract()
  1018. # print('删除表格业绩内容', del_tag.text)
  1019. def is_all_winner(title):
  1020. '''
  1021. 是否提取所有投标人作为中标人,存管类不分排名都作中标人;入围类按排名,无排名都做中标人
  1022. :param title: 标题
  1023. :return:
  1024. '''
  1025. if re.search('(资金|公款|存款)?竞争性存[放款]|(资金|公款|存款)存放|存放银行|存款服务|国库现金管理', title):
  1026. return 1
  1027. elif re.search('招募|入围|框架采购|(单位|商|机构)入库|入库供应商', title):
  1028. return 2
  1029. return False
  1030. def is_deposit_project(title, name, requirement):
  1031. '''
  1032. 通过正则判断项目是否为银行存款类项目
  1033. :param title: 标题
  1034. :param name: 项目名称
  1035. :param requirement: 采购内容
  1036. :return:
  1037. '''
  1038. if re.search('(资金|公款|存款)?竞争性存[放款]|(资金|公款|存款)((.{2,10}))?存放|存放银行|存款(服务|业务|项目)|国库现金管理|存款账户开户|(管理|存款|合作)(定点|专户)?银行|贷款合作银行|资金监管账户|开户银行项目|专户开户银行|银行专户选择|定期存[款放]|专项债券?专用账户', title+name+requirement):
  1039. return True
  1040. return False
  1041. def get_money_entity(sentence_text, found_yeji=0, in_attachment=False):
  1042. money_list = []
  1043. # 使用正则识别金额
  1044. entity_type = "money"
  1045. list_money_pattern = {"cn": "(()(?P<filter_kw>百分之)?(?P<money_cn>[零壹贰叁肆伍陆柒捌玖拾佰仟萬億圆十百千万亿元角分]{3,})())",
  1046. "key_word": "((?P<text_key_word>(?:[¥¥]+,?|(中标|成交|合同|承租|投资|服务))?(金?额|价格?)|价格|预算(金额)?|(监理|设计|勘察)(服务)?费|[单报标限总造]价款?|金额|租金|标的基本情况|CNY|成交结果|资金|(控制|拦标)价|投资)(\d:|\d=\d[-+×]\d:)?(?:[,,\[(\(]*\s*(人民币|单位:)?/?(?P<unit_key_word_before>[万亿]?(?:[美日欧]元|元(/(M2|[\u4e00-\u9fa5]{1,3}))?)?(?P<filter_unit2>[台个只吨]*))\s*(/?费率)?(人民币)?[\])\)]?)\s*[,,::]*(RMB|USD|EUR|JPY|CNY)?[::]?(\s*[^壹贰叁肆伍陆柒捌玖拾佰仟萬億分万元编号时间日期计采a-zA-Z]{,8}?))(第[123一二三]名[::])?(\d+(\*\d+%)+=)?(?P<money_key_word>\d+,\d+\.\d{2,6}|\d{1,3}([,,]\d{3})+(\.\d+)?|\d+(\.\d+)?[百千]{,1})(?P<science_key_word>(E-?\d+))?(?:[(\(]?(?P<filter_>[%%‰折])*\s*,?((金额)?单位[::])?(?P<unit_key_word_behind>[万亿]?(?:[美日欧]元|元)?(?P<filter_unit1>[台只吨斤棵株页亩方条天年月日]*))\s*[)\)]?))",
  1047. "front_m": "((?P<text_front_m>(?:[(\(]?\s*(?P<unit_front_m_before>[万亿]?(?:[美日欧]元|元))\s*[)\)]?)\s*[,,::]*(\s*[^壹贰叁肆伍陆柒捌玖拾佰仟萬億分万元编号时间日期计采a-zA-Z金额价格]{,2}?))(?P<money_front_m>\d{1,3}([,,]\d{3})+(\.\d+)?|\d+(\.\d+)?(?:,?)[百千]*)(?P<science_front_m>(E-?\d+))?())",
  1048. "behind_m": "(()()(?P<money_behind_m>\d{1,3}([,,]\d{3})+(\.\d+)?|\d+(\.\d+)?(?:,?)[百千]*)(?P<science_behind_m>(E-?\d+))?(人民币)?[\((]?(?P<unit_behind_m>[万亿]?(?:[美日欧]元|元)(?P<filter_unit3>[台个只吨斤棵株页亩方条米]*))[\))]?)"}
  1049. # 2021/7/19 调整金额,单位提取正则,修复部分金额因为单位提取失败被过滤问题。 20240415 调整front_m 修复 详见合同元,合同金额:378.8万元 提取
  1050. pattern_money = re.compile("%s|%s|%s|%s" % (
  1051. list_money_pattern["cn"], list_money_pattern["key_word"], list_money_pattern["behind_m"],
  1052. list_money_pattern["front_m"]))
  1053. # sentence_text = re.sub('\d+[年月日]', '', sentence_text) # 修复560180018 中标价(元):3年投标报价(元)含税6299700.00 3年作为金额
  1054. if re.search('业绩(公示|汇总|及|报告|\w{,2}(内容|情况|信息)|[^\w])', sentence_text):
  1055. found_yeji += 1
  1056. if found_yeji >= 2: # 过滤掉业绩后面的所有金额
  1057. all_match = []
  1058. else:
  1059. ser = re.search('((收费标准|计算[方公]?式):|\w{3,5}\s*=)+\s*[中标投标成交金额招标人预算价格万元\s()()\[\]【】\d\.%%‰\+\-*×/]{20,}[,。]?', sentence_text) # 过滤掉收费标准里面的金额
  1060. if ser:
  1061. sentence_text = sentence_text.replace(ser.group(0), ' ' * len(ser.group(0)))
  1062. all_match = re.finditer(pattern_money, sentence_text)
  1063. # print('all_match:', all_match)
  1064. for _match in all_match:
  1065. # print('_match: ', _match.group())
  1066. if re.search('^元/1\d{10},$', _match.group(0)): # 修复 495042766 现场负责人 姚元 / 13488160460 预测为金额
  1067. continue
  1068. if len(_match.group()) > 0:
  1069. # print("===",_match.group())
  1070. # # print(_match.groupdict())
  1071. notes = '' # 2021/7/20 新增备注金额大写或金额单位 if 金额大写 notes=大写 elif 单位 notes=单位
  1072. unit = ""
  1073. entity_text = ""
  1074. start_index = ""
  1075. end_index = ""
  1076. text_beforeMoney = ""
  1077. filter = ""
  1078. filter_unit = False
  1079. notSure = False
  1080. science = ""
  1081. if re.search('业绩(公示|汇总|及|报告|\w{,2}(内容|情况|信息)|[^\w])', sentence_text[:_match.span()[0]]): # 2021/7/21过滤掉业绩后面金额
  1082. # print('金额在业绩后面: ', _match.group(0))
  1083. found_yeji += 1
  1084. break
  1085. for k, v in _match.groupdict().items():
  1086. if v != "" and v is not None:
  1087. if k == 'text_key_word':
  1088. notSure = True
  1089. if k.split("_")[0] == "money":
  1090. entity_text = v
  1091. # print(_match.group(k), 'entity_text: ', sentence_text[_match.start(k): _match.end(k)])
  1092. if entity_text.endswith(',00'): # 金额逗号后面不可能为两个0结尾,应该小数点识别错,直接去掉
  1093. entity_text = entity_text[:-3]
  1094. if k.split("_")[0] == "unit":
  1095. if 'behind' in k or unit == "": # 优先后面单位 预算金额(元):160万元 总价(万元):最终报价:695000.00(元)
  1096. unit = v
  1097. if k.split("_")[0] == "text":
  1098. text_beforeMoney = v
  1099. if k.split("_")[0] == "filter":
  1100. filter = v
  1101. if re.search("filter_unit", k) is not None:
  1102. filter_unit = True
  1103. if k.split("_")[0] == 'science':
  1104. science = v
  1105. # print("金额:{0} ,单位:{1}, 前文:{2}, filter: {3}, filter_unit: {4}".format(entity_text,unit,text_beforeMoney,filter,filter_unit))
  1106. # if re.search('(^\d{2,},\d{4,}万?$)|(^\d{2,},\d{2}万?$)', entity_text.strip()): # 2021/7/19 修正OCR识别小数点为逗号
  1107. # if re.search('[幢栋号楼层]', sentence_text[max(0, _match.span()[0] - 2):_match.span()[0]]):
  1108. # entity_text = re.sub('\d+,', '', entity_text)
  1109. # else:
  1110. # entity_text = entity_text.replace(',', '.')
  1111. # # print(' 修正OCR识别小数点为逗号')
  1112. if filter != "":
  1113. continue
  1114. if len(entity_text)>30 or len(re.sub('[E-]', '', science))>2: # 限制数字长度,避免类似265339018附件金额错误,数值超大报错 decimal.InvalidOperation
  1115. continue
  1116. start_index, end_index = _match.span()
  1117. start_index += len(text_beforeMoney)
  1118. '''过滤掉手机号码作为金额'''
  1119. if re.search('电话|手机|联系|方式|编号|编码|日期|数字|时间', text_beforeMoney):
  1120. # print('过滤掉手机号码作为金额')
  1121. continue
  1122. elif re.search('^1[3-9]\d{9}$', entity_text) and re.search(':\w{1,3}$', text_beforeMoney): # 过滤掉类似 '13863441880', '金额(万元):季勇13863441880'
  1123. # print('过滤掉手机号码作为金额')
  1124. continue
  1125. elif re.search('^\d(.\d{1,2})?$', entity_text) and re.search('\d$', _match.group(0)) and re.search('^[、.]', sentence_text[_match.end():]): # 170756755 控制价为:1、合理利润率上限
  1126. # print('过滤错误金额:', _match.group(0))
  1127. continue
  1128. if unit == "": # 2021/7/21 有明显金额特征的补充单位,避免被过滤
  1129. if (re.search('(¥|¥|RMB|CNY)[::]?$', text_beforeMoney) or re.search('[零壹贰叁肆伍陆柒捌玖拾佰仟萬億圆十百千万亿元角分]{3,}', entity_text)):
  1130. if entity_text.endswith('万元'):
  1131. unit = '万元'
  1132. entity_text = entity_text[:-2]
  1133. else:
  1134. unit = '元'
  1135. # print('1明显金额特征补充单位 元')
  1136. elif re.search('USD[::]?$', text_beforeMoney):
  1137. unit = '美元'
  1138. elif re.search('EUR[::]?$', text_beforeMoney):
  1139. unit = '欧元'
  1140. elif re.search('JPY[::]?$', text_beforeMoney):
  1141. unit = '日元'
  1142. elif re.search('^[-—]+[\d,.]+万元', sentence_text[end_index:]):
  1143. # print('两个金额连接后面的有单位,用后面单位')
  1144. unit = '万元'
  1145. elif re.search('^,?(价格币种:\w{2,3},)?价格单位:万元', sentence_text[end_index:]): # 修复494731937金额单位缺漏 中标价格:39501.094425,价格币种:人民币,价格单位:万元,
  1146. unit = '万元'
  1147. elif re.search('万元', sentence_text[max(0, start_index-10):start_index]): #修复511402017 价格类型:(万元)报价:13311.1582,得分:84.46,
  1148. unit = '万元'
  1149. elif re.search('([单报标限总造]价款?|金额|租金|(中标|成交|合同|承租|投资|控制|拦标))?[价额]|价格|预算(金额)?|(监理|设计|勘察)(服务)?费)(小写)?[::为]*-?$', text_beforeMoney.strip()) and re.search('^0|1[3|4|5|6|7|8|9]\d{9}', entity_text) == None: # 修复
  1150. if re.search('^[\d,,.]+$', entity_text) and float(re.sub('[,,]', '', entity_text))<500 and re.search('万元', sentence_text):
  1151. unit = '万元'
  1152. # print('金额较小且句子中有万元的,补充单位为万元')
  1153. elif re.search('^\d{1,3}\.\d{4,6}$', entity_text) and re.search('0000$', entity_text) == None:
  1154. unit = '万元'
  1155. else:
  1156. unit = '元'
  1157. # print('金额前面紧接关键词的补充单位 元')
  1158. elif re.search('(^\d{,3}(,?\d{3})+(\.\d{2,7},?)$)|(^\d{,3}(,\d{3})+,?$)', entity_text):
  1159. unit = '元'
  1160. # print('3明显金额特征补充单位 元')
  1161. else:
  1162. # print('过滤掉没单位金额: ',entity_text)
  1163. continue
  1164. elif unit == '万元':
  1165. if end_index < len(sentence_text) and sentence_text[end_index] == '元' and re.search('\d$', entity_text):
  1166. unit = '元'
  1167. elif re.search('^[5-9]\d{6,}\.\d{2}$', entity_text): # 五百亿以上的万元改为元
  1168. unit = '元'
  1169. if unit.find("万") >= 0 and entity_text.find("万") >= 0: # 2021/7/19修改为金额文本有万,不计算单位
  1170. # print('修正金额及单位都有万, 金额:',entity_text, '单位:',unit)
  1171. unit = "元"
  1172. if re.search('.*万元万元', entity_text): # 2021/7/19 修正两个万元
  1173. # print(' 修正两个万元',entity_text)
  1174. entity_text = entity_text.replace('万元万元', '万元')
  1175. else:
  1176. if filter_unit:
  1177. continue
  1178. # symbol = '-' if entity_text.startswith('-') and not entity_text.startswith('--') and re.search('\d+$', sentence_text[:begin_index_temp]) == None else '' # 负值金额前面保留负号 ,后面这些不作为负金额 起拍价:105.29-200.46万元 预 算 --- 350000.0 2023/04/14 取消符号
  1179. entity_text = re.sub("[^0-9.零壹贰叁肆伍陆柒捌玖拾佰仟萬億圆十百千万亿元角分]", "", entity_text)
  1180. # print('转换前金额:', entity_text, '单位:', unit, '备注:',notes, 'text_beforeMoney:',text_beforeMoney)
  1181. if re.search('总投资|投资总额|总预算|总概算|(投资|招标|资金|存放|操作|融资)规模|批复概算|投资额|总规模|工程造价|总金额',
  1182. sentence_text[max(0, _match.span()[0] - 10):_match.span()[1]]): # 2021/8/5过滤掉总投资金额 20241031工程造价作总投资
  1183. # print('总投资金额: ', _match.group(0))
  1184. notes = '总投资'
  1185. elif re.search('投资|概算|建安费|其他费用|基本预备费',
  1186. sentence_text[max(0, _match.span()[0] - 8):_match.span()[1]]): # 2021/11/18 投资金额不作为招标金额
  1187. notes = '投资'
  1188. # elif re.search('工程造价',
  1189. # sentence_text[max(0, _match.span()[0] - 8):_match.span()[1]]): # 2021/12/20 工程造价不作为招标金额
  1190. # notes = '工程造价'
  1191. elif (re.search('保证金', sentence_text[max(0, _match.span()[0] - 5):_match.span()[1]])
  1192. or re.search('保证金的?(缴纳)?(金额|金\?|额|\?)?[\((]*(万?元|为?人民币|大写|调整|变更|已?修改|更改|更正)?[\))]*[::为]',
  1193. sentence_text[max(0, _match.span()[0] - 10):_match.span()[1]])
  1194. or re.search('保证金由[\d.,]+.{,3}(变更|修改|更改|更正|调整?)为',
  1195. sentence_text[max(0, _match.span()[0] - 15):_match.span()[1]])):
  1196. notes = '保证金'
  1197. # print('保证金信息:', sentence_text[max(0, _match.span()[0] - 15):_match.span()[1]])
  1198. elif re.search('成本(警戒|预警)(线|价|值)[^0-9元]{,10}',
  1199. sentence_text[max(0, _match.span()[0] - 10):_match.span()[0]]):
  1200. notes = '成本警戒线'
  1201. elif re.search('(监理|设计|勘察)(服务)?费(报价)?[约为:]|服务金额', sentence_text[_match.span()[0]:_match.span()[1]]):
  1202. # cost_re = re.search('(监理|设计|勘察)(服务)?费', sentence_text[_match.span()[0]:_match.span()[1]])
  1203. # notes = cost_re.group(1)
  1204. notes = '招标或中标金额'
  1205. elif re.search('单价|总金额', sentence_text[_match.span()[0]:_match.span()[1]]):
  1206. notes = '单价'
  1207. elif re.search('^[/每]', sentence_text[_match.end():]):
  1208. # print('单价:', _match.group(0))
  1209. notes = '单价'
  1210. elif re.search('[零壹贰叁肆伍陆柒捌玖拾佰仟萬億圆]', entity_text) != None:
  1211. notes = '大写'
  1212. if entity_text[0] == "拾": # 2021/12/16 修正大写金额省略了数字转换错误问题
  1213. entity_text = "壹" + entity_text
  1214. # print("补充备注:notes = 大写")
  1215. if len(unit) > 0:
  1216. if unit.find('万') >= 0 and len(entity_text.split('.')[0]) >= 8: # 2021/7/19 修正万元金额过大的情况
  1217. # print('修正单位万元金额过大的情况 金额:', entity_text, '单位:', unit)
  1218. entity_text = str(
  1219. getUnifyMoney(entity_text) * getMultipleFactor(re.sub("[美日欧]", "", unit)[0]) / 10000)
  1220. unit = '元' # 修正金额后单位 重置为元
  1221. else:
  1222. # print('str(getUnifyMoney(entity_text)*getMultipleFactor(unit[0])):')
  1223. entity_text = str(getUnifyMoney(entity_text) * getMultipleFactor(re.sub("[美日欧]", "", unit)[0]))
  1224. else:
  1225. if entity_text.find('万') >= 0 and entity_text.split('.')[0].isdigit() and len(
  1226. entity_text.split('.')[0]) >= 8:
  1227. entity_text = str(getUnifyMoney(entity_text) / 10000)
  1228. # print('修正金额字段含万 过大的情况')
  1229. else:
  1230. entity_text = str(getUnifyMoney(entity_text))
  1231. if science and re.search('^E-?\d+$', science): # 科学计数
  1232. entity_text = str(Decimal(entity_text + science)) if Decimal(entity_text + science) > 100 and Decimal(
  1233. entity_text + science) < 10000000000 else entity_text # 结果大于100及小于100万才使用科学计算
  1234. if float(entity_text) > 100000000000: # float(entity_text)<100 or 2022/3/4 取消最小金额限制
  1235. # print('过滤掉金额:float(entity_text)<100 or float(entity_text)>100000000000', entity_text, unit)
  1236. continue
  1237. if notSure and unit == "" and float(entity_text) > 100 * 10000:
  1238. # print('过滤掉金额 notSure and unit=="" and float(entity_text)>100*10000:', entity_text, unit)
  1239. continue
  1240. # print("金额:{0} ,单位:{1}, 前文:{2}, filter: {3}, filter_unit: {4}".format(entity_text, unit, text_beforeMoney,
  1241. # filter, filter_unit))
  1242. if re.search('[%%‰折]|费率|下浮率', text_beforeMoney) and float(entity_text)<1000: # 过滤掉可能是费率的金额
  1243. # print('过滤掉可能是费率的金额')
  1244. continue
  1245. money_list.append((entity_text, start_index, end_index, unit, notes))
  1246. return money_list, found_yeji
  1247. def recall(y_true, y_pred):
  1248. '''
  1249. 计算召回率
  1250. @Argus:
  1251. y_true: 正确的标签
  1252. y_pred: 模型预测的标签
  1253. @Return
  1254. 召回率
  1255. '''
  1256. c1 = K.sum(K.round(K.clip(y_true * y_pred, 0, 1)))
  1257. c3 = K.sum(K.round(K.clip(y_true, 0, 1)))
  1258. if c3 == 0:
  1259. return 0
  1260. recall = c1 / c3
  1261. return recall
  1262. def f1_score(y_true, y_pred):
  1263. '''
  1264. 计算F1
  1265. @Argus:
  1266. y_true: 正确的标签
  1267. y_pred: 模型预测的标签
  1268. @Return
  1269. F1值
  1270. '''
  1271. c1 = K.sum(K.round(K.clip(y_true * y_pred, 0, 1)))
  1272. c2 = K.sum(K.round(K.clip(y_pred, 0, 1)))
  1273. c3 = K.sum(K.round(K.clip(y_true, 0, 1)))
  1274. precision = c1 / c2
  1275. if c3 == 0:
  1276. recall = 0
  1277. else:
  1278. recall = c1 / c3
  1279. f1_score = 2 * (precision * recall) / (precision + recall)
  1280. return f1_score
  1281. def precision(y_true, y_pred):
  1282. '''
  1283. 计算精确率
  1284. @Argus:
  1285. y_true: 正确的标签
  1286. y_pred: 模型预测的标签
  1287. @Return
  1288. 精确率
  1289. '''
  1290. c1 = K.sum(K.round(K.clip(y_true * y_pred, 0, 1)))
  1291. c2 = K.sum(K.round(K.clip(y_pred, 0, 1)))
  1292. precision = c1 / c2
  1293. return precision
  1294. # def print_metrics(history):
  1295. # '''
  1296. # 制作每次迭代的各metrics变化图片
  1297. #
  1298. # @Arugs:
  1299. # history: 模型训练迭代的历史记录
  1300. # '''
  1301. # import matplotlib.pyplot as plt
  1302. #
  1303. # # loss图
  1304. # loss = history.history['loss']
  1305. # val_loss = history.history['val_loss']
  1306. # epochs = range(1, len(loss) + 1)
  1307. # plt.subplot(2, 2, 1)
  1308. # plt.plot(epochs, loss, 'bo', label='Training loss')
  1309. # plt.plot(epochs, val_loss, 'b', label='Validation loss')
  1310. # plt.title('Training and validation loss')
  1311. # plt.xlabel('Epochs')
  1312. # plt.ylabel('Loss')
  1313. # plt.legend()
  1314. #
  1315. # # f1图
  1316. # f1 = history.history['f1_score']
  1317. # val_f1 = history.history['val_f1_score']
  1318. # plt.subplot(2, 2, 2)
  1319. # plt.plot(epochs, f1, 'bo', label='Training f1')
  1320. # plt.plot(epochs, val_f1, 'b', label='Validation f1')
  1321. # plt.title('Training and validation f1')
  1322. # plt.xlabel('Epochs')
  1323. # plt.ylabel('F1')
  1324. # plt.legend()
  1325. #
  1326. # # precision图
  1327. # prec = history.history['precision']
  1328. # val_prec = history.history['val_precision']
  1329. # plt.subplot(2, 2, 3)
  1330. # plt.plot(epochs, prec, 'bo', label='Training precision')
  1331. # plt.plot(epochs, val_prec, 'b', label='Validation pecision')
  1332. # plt.title('Training and validation precision')
  1333. # plt.xlabel('Epochs')
  1334. # plt.ylabel('Precision')
  1335. # plt.legend()
  1336. #
  1337. # # recall图
  1338. # recall = history.history['recall']
  1339. # val_recall = history.history['val_recall']
  1340. # plt.subplot(2, 2, 4)
  1341. # plt.plot(epochs, recall, 'bo', label='Training recall')
  1342. # plt.plot(epochs, val_recall, 'b', label='Validation recall')
  1343. # plt.title('Training and validation recall')
  1344. # plt.xlabel('Epochs')
  1345. # plt.ylabel('Recall')
  1346. # plt.legend()
  1347. #
  1348. # plt.show()
  1349. if __name__=="__main__":
  1350. # print(fool_char_to_id[">"])
  1351. print(getUnifyMoney('伍仟贰佰零壹拾伍万零捌佰壹拾元陆角伍分'))
  1352. # model = getModel_w2v()
  1353. # vocab,matrix = getVocabAndMatrix(model, Embedding_size=128)
  1354. # save([vocab,matrix],"vocabMatrix_words.pk")