channel.py 60 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854
  1. # -*- coding: utf-8 -*-
  2. """``DocChannel`` — 公告类型/生命周期模型+规则混合预测。
  3. Phase 5 从 ``interface/predictor.py``(约 4722-5551 行)迁出。
  4. 原 ``from common.Utils import *`` / ``from common.nerUtils import *``
  5. 已替换为显式 import;``os.path.dirname(__file__)`` 路径引用替换为
  6. ``predictors._common.INTERFACE_DIR``。
  7. """
  8. from __future__ import absolute_import
  9. import os
  10. import re
  11. import json
  12. import numpy as np
  13. import tensorflow as tf
  14. from BiddingKG.dl.common.logging import log
  15. from BiddingKG.dl.foolnltk import selffool
  16. from BiddingKG.dl.model_runtime.vocab import getIndexOfWords
  17. from BiddingKG.dl.model_runtime.embed import embedding
  18. from BiddingKG.dl.predictors._common import INTERFACE_DIR
  19. __all__ = ["DocChannel"]
  20. class DocChannel():
  21. def __init__(self, life_model='channel_savedmodel/channel.pb', type_model='channel_savedmodel/doctype.pb',config=None):
  22. self.lift_sess, self.lift_title, self.lift_content, self.lift_prob, self.lift_softmax,\
  23. self.mask, self.mask_title = self.load_life(life_model,config)
  24. self.type_sess, self.type_title, self.type_content, self.type_prob, self.type_softmax,\
  25. self.type_mask, self.type_mask_title = self.load_type(type_model)
  26. self.sequen_len = 200 # 150 200
  27. self.title_len = 30
  28. self.sentence_num = 10
  29. self.kws = '供货商|候选人|供应商|入选人|项目|选定|预告|中标|成交|补遗|延期|报名|暂缓|结果|意向|出租|补充|合同|限价|比选|指定|工程|废标|取消|中止|流标|资质|资格|地块|招标|采购|货物|租赁|计划|宗地|需求|来源|土地|澄清|失败|探矿|预审|变更|变卖|遴选|撤销|意见|恢复|采矿|更正|终止|废置|报建|流拍|供地|登记|挂牌|答疑|中选|受让|拍卖|竞拍|审查|入围|更改|条件|洽谈|乙方|后审|控制|暂停|用地|询价|预'
  30. lb_type = ['采招数据', '土地矿产', '拍卖出让', '产权交易', '新闻资讯']
  31. lb_life = ['采购意向', '招标预告', '招标公告', '招标答疑', '公告变更', '资审结果', '中标信息', '合同公告', '废标公告']
  32. self.id2type = {k: v for k, v in enumerate(lb_type)}
  33. self.id2life = {k: v for k, v in enumerate(lb_life)}
  34. self.load_pattern()
  35. def load_pattern(self):
  36. self.type_dic = {
  37. '土地矿产': '供地结果|(土地|用地|宗地|地块|海域|矿)的?(基本信息|基本情况|概况|信息|详情|来源|用途|性质|编号|位置|坐落|使用年限|出让年限)|(土地|山地|农田)(经营权)?(出让|出租|招租|租赁|承包|流转)|流转土地',
  38. '拍卖出让': '(拍卖|变卖|流拍|竞拍|竞买)的?(公告|活动|信息|结果|成交|主体|标的|资产|财产|方式|类型|流程|程序|规则|价格|保证金|时间)|(公开|进行|密封)(拍卖|变卖|竞拍)|第[一二三四五六七八九\d]次拍卖|(资产|司法|网络)拍卖|交易方式.{,2}拍卖|拍卖会|(拍卖.?方式|起拍价)[::]|竞买人资格|竞买人资质要求',
  39. '产权交易': '(产权|资产|权证)的?(类型|类别|用途|性质|状态|信息|名称|编号|(基本)?情况)|(经营权|承包权|使用权|租赁权|股权|债权|排污权|化学需氧量|储备量)(挂牌|转让|出让)|竞价销售|销售结果|房屋所有权房产|免租期限|交易期限|(受让|转让|承租|出租)(人|方)|(店面|店铺|商铺|铺位?|门面|门市|食堂|饭堂|校舍|车位|停车场|厂?房|仓?库|馆|资产|物业|房产|房屋|场地|农田|鱼?塘)\w{,4}(处置|招租|出租|续租|租赁|转让)|(出租|转让|产权|资产)(项目|中标|成交|流标|废标)|出租(用途|类型)|转让底价|租赁(标的物|情况)|看[样货](时间|地[点址]|方式|仓库|验货)|最小加价|加价[幅梯]度|交易模式[::\s]*延时竞价销售|挂牌(开始|结束)时间|挂牌价格?',
  40. '采招数据': '(采购|招标)(条件|范围|文件|内容)|(申请人|投标人|供应商|报价人|参选人)的?资格要求;|采购需求清单|最低价排序|竞争性采购方式|采购进行公开竞价|竞价模式[::\s]*一次报价|预算金额|代理银行资格选定' # |变更|答疑|澄清|中标|成交|合同|废标|流标 |(采购|招标|代理)(人|机构|单位)|
  41. }
  42. self.title_type_dic = { # ‘**2’ 为类别强相关关键词,不会被其他规则修正
  43. '土地矿产': '(土地|用地|宗地|荒地|山地|海域|矿)(出让|出租|招租|租赁|承包|流转|使用权|经营权|征收|划拨|中标|成交)|供地结果|矿业权|探矿权|采矿权|(土地|用地|宗地|地块)(使用权)?(终止|中止|网上)?(挂牌|出让|拍卖|招拍|划拨)|征收土地|流转土地',
  44. '土地矿产2': '(土地|用地|宗地|荒地|山地|海域|矿)(出让|出租|招租|租赁|承包|流转|使用权|经营权|征收|划拨|中标|成交)|供地结果|矿业权|探矿权|采矿权|(土地|用地|宗地|地块)(使用权)?(终止|中止|网上)?(挂牌|出让|拍卖|招拍|划拨)|征收土地|流转土地',
  45. '拍卖出让': '(拍卖|变卖|拍(变)卖|流拍|竞拍|竞买)[\)\])】]?的?(公告|公示|预告|告知书)|拍卖|变卖|流拍|竞拍|第[一二三四五六七八九\d]次拍卖|拍卖会$',
  46. '拍卖出让2': '(拍卖|变卖|拍(变)卖|流拍|竞拍|竞买)[\)\])】]?的?(公告|公示|预告|告知书)|拍卖|变卖|流拍|第[一二三四五六七八九\d]次拍卖|拍卖会$',
  47. '产权交易': '经营权|承包权|使用权|租赁权|股权|债权|排污权|化学需氧量|储备量|竞价销售|销售结果|出租|招租|拍租|竞租|续租|挂牌|出让|废[旧弃]?(物资|设备|资源|金属|钢筋|料)处[置理]',
  48. '产权交易2': '使用权|租赁权|股权|债权|排污权|竞价销售|销售结果|出租|招租|拍租|竞租|续租|挂牌|出让|废[旧弃]?(物资|设备|资源|金属|钢筋|料)处[置理]',
  49. # '采招数据': '(采购|招标|询价|议价|比价|比选|遴选|邀请|邀标|磋商|洽谈|约谈|谈判|征询|调研)的?(公告|公示|中标|成交|结果|$)|工程招标|定点服务|(设备|服务|\w{2})[直采]购|(建设|改造)项目|工程|拦标价|控制价|银行|资格选定|资金|公款|存款|存放|现金管理|招募|入围|入库',
  50. '采招数据': '(采购|招标|询价|议价|比价|比选|遴选|邀请|邀标|磋商|洽谈|约谈|谈判|征询|调研)的?(公告|公示|中标|成交|结果|$)|工程招标|定点服务|(设备|服务|\w{2})[直采]购|(建设|改造)项目|拦标价|控制价|资格选定|资格认定|资金|公款|存款|现金管理|招募|入库|遴选.{,25}(服务|事务所|机构)',
  51. # |竞价 采招/产权都有竞价方式 # 意向|需求|预公?告|报建|总承包|工程|施工|设计|勘察|代理|监理 |变更|答疑|澄清|中标|成交|合同|废标|流标
  52. '新闻资讯': '(考试|面试|笔试)成绩|入围面试|面试人员名单|成绩的?(公告|公示|公布)|公开招聘|招聘(公告|简章|启事|合同制)|疫情防控\s{,5}(通知|情况|提示)|行政审批结果|收回收购(存量)?(闲置)?土地.{,3}价格.{,2}公示',
  53. '新闻资讯2': '面试人员名单|疫情防控\s{,5}(通知|情况|提示)|行政审批结果|收回收购(存量)?(闲置)?土地.{,3}价格.{,2}公示|(印发|转发)《.{2,30}》.{,3}通知|清退.{,8}保证金|专项资金管理清单.{,2}使用情况|关于依法处置'
  54. }
  55. self.life_dic = {
  56. '采购意向': '采购意向|招标意向|选取意向|意向公告|意向公示',
  57. '采购意向neg': '发布政府采购意向|采购意向公告已于',
  58. '招标预告': '(预计|计划)(招标|采购|发标|发包)(时间|日期)|采购(计划编号|需求方案|预告|预案)|(预|需求)公示|需求(方案|信息|论证|公告|公示)',
  59. '招标公告': '(采购|招标|竞选|报名)条件|报名(时间|流程|方法|要求|\w{,5}材料)[:\s]|[^\w]成交规则|参加竞价采购交易资格|(申请人|投标人|供应商|报价人|参选人)的?资格(要求|条件)|获取(采购|招标|询价|议价|竞价|比价|比选|遴选|邀请|邀标|磋商|洽谈|约谈|谈判|竞谈|应答)文件|(采购|招标|询价|议价|竞价|比价|比选|遴选|邀请|邀标|磋商|洽谈|约谈|谈判|竞谈|应答)文件的?(获取|领取)|评选方式:?\s*价格最低|求购信息',
  60. '资审结果': '资审及业绩公示|资审结果及业绩|资格后审情况报告|资格(后审|预审|审查)结果(公告|公示)|(预审|审查)工作已经?结束|未通过原因', #|资格
  61. '招标答疑': '现澄清(为|如下)|答疑补遗|澄清内容如下|第[0-9一二三四五]次澄清|答疑澄清|(最高(投标)?限价|控制价|拦标价)公示|澄清与修改', # |异议的回复
  62. '公告变更': '第[\d一二]次变更|(更正|变更)(公告|公示|信息|内容|事项|原因|理由|日期|时间|如下)|原公告((主要)?(信息|内容)|发布时间)|(变更|更正)[前后]内容|现?在?(变更|更正|修改|更改)(内容)?为|(公告|如下|信息|内容|事项|结果|文件|发布|时间|日期)(更正|变更)|恢复[\u4e00-\u9fa5]{,8}(招标?投标|招标|投标)活动',
  63. '公告变更neg': '履约变更内容',
  64. '候选人公示': '候选人公示|评标结果公示|中标候选人名单公示|现将中标候选人(进行公示|公[示布]如下)|(中标|中选)候选人[\u4e00-\u9fa5]{0,2}(信息|情况|第[123一二三]名)[::\s]',
  65. '候选人公示neg': '中标候选人公示期|中标候选人公示前',
  66. '中标信息': '供地结果信息|采用单源直接采购的?情况说明|[特现]?将\w{,4}(成交|中标|中选|选定结果|选取结果|入围结果|竞价结果)\w{,4}(进行公示|公[示布]如下)|(询价|竞价|遴选)?(成交|中标|中选)(公告|公示)|(成交|中标|中选|选定|选取|入围|询价)结果(如下|公告|公示)|(中标|中选)(供应商|承包商|候选人|入围单位)如下|拟定供应商的情况|((中标|中选)(人|成交)|成交)\w{,3}(信息|情况)[::\s]',
  67. '中标信息2': '\s(成交|中标|中选)(信息|日期|时间|总?金额|价格)[::\s]|(成交|中标|中选)价格\s*[\d.,]+(?万?元|(采购|招标|成交|中标|中选|评标)结果|单一来源(采购|招标)?的?(中标|成交|结果)|项目已结束|中标公示 ', # |单一来源采购原因|拟采取单一来源方式采购|单一来源采购公示
  68. '中标信息3': '(中标|中选|成交|拟定|拟选用|最终选定的?|受让)(供应商|供货商|服务商|机构|企业|公司|单位|候选人|人)(信息[,:]?)?(名称)?[::\s]|[、\s](第一名|(拟定|推荐|入围)?(供应商|供货商)|(中选|中标|供货)单位|中选人)[::\s]|确定[\w()]{6,25}为中标人', # |唯一
  69. '中标信息neg': '按项目控制价下浮\d%即为成交价|成交原则|不得确定为(中标|成交)|招标人按下列原则选择中标人|评选成交供应商:|拟邀请供应商|除单一来源采购项目外|单一来源除外|(各.{,5}|尊敬的)(供应商|供货商)[:\s]|竞拍起止时间:|询价结果[\s\n::]*不公开|本项目已具备招标条件|现对该项目进行招标公告|发布\w{2}结果后\d天内送达|本次\w{2}结果不对外公示|供应商\s*资格要求|成交情况:\s*[流废]标|中标单位:本次招标拟?中标单位\d家|通知中标单位|影响(成交|中标)结果|确定为成交供应商|(成交|中标|中选)公[告示](发布|\w{,2})后|竞价成交后', # 503076535 按照服务方案的优劣 确定为成交供应商
  70. # |确定成交供应商[:,\s]
  71. '合同公告': '合同(公告|公示|信息|内容)|合同(编号|名称|主体|基本情况|完成(日期|时间))|(供应商乙方|乙方供应商):|合同总?金额|履约信息',
  72. '合同公告neg': '合同金额页|合同金额[\u4e00-\u9fa5]{,2}不[少低高多]于',
  73. '废标公告': '(终止|中止|废标|流标|流采|失败|作废|异常|撤销)(结果)?(公告|公示|招标|采购|竞价)|(谈判结果为|结果类型|采购结果):?(废标|流标)|((本|该)(项目|标段|合同|合同包|采购包|次)\w{,5})((失败|终止|流标|废标)|予以(废标|流标)|(按|做|作)?(流标|废标|废置)处理)|(采购|招标|询价|议价|竞价|比价|比选|遴选|邀请|邀标|磋商|洽谈|约谈|谈判|竞谈|应答|项目)(终止|中止|废标|流标|失败|作废|异常|撤销)|(废标|流标)处理',
  74. '废标公告2': '(无效|中止|终止|废标|流标|失败|作废|异常|撤销)([\((](无效|中止|终止|废标|流标|失败|作废|异常|撤销)[\))])?的?(原因|理由)|本项目因故取消|本(项目|次)(公开)?\w{2}失败|已终止\s*原因:|(人|人数|供应商|单位)(不足|未达\w{,3}数量)|已终止|不足[3三]家|无(废标)|成交情况:\s*[流废]标|现予以废置|报名未够三家',
  75. '废标公告neg': '超过此报价将作为[废流]标处理|否则按[废流]标处理|终止规则:|成交规则:|视为流标|竞价失败的一切其他情形|是否废标:否|若不足三家公司参与|供应商数量:?\s*报名供应商不足三家|有效报价不足三家,\s*系统自动废标|如遇项目流[标采]|[如若][\u4e00-\u9fa5]{,10}(不足|不够|未够|未达)[3三]家' # 503076535 供应商数量: 报名供应商不足三家。
  76. }
  77. self.title_life_dic = {
  78. '采购意向': '采购意向|招标意向|选取意向|意向公告|意向公示|意向公开',
  79. '招标预告': '预公?告|预公示|报建公告|(批前|标前)公示|(供应|招标)计划(表|书|任务|公告|公示|发布)?$|[\[【](供应|招标)计划[\]】]|(论证|征求|征集)(供应商)?意见|意见征询|需求评审公告|需求(公告|公示|意见)',
  80. '公告变更': '第[\d一二]次变更|(变更|更正(事项)?|更改|延期|暂停|恢复)(招标|采购)?的?(公告|公示|通知)|变更$|更正$',
  81. '招标答疑': '质疑|澄清|答疑(文件)?|补遗书?|(最高(投标)?限价|控制价|拦标价)(公示|公告|$)|补充说明',
  82. '废标公告': '(终止|中止|废标|废除|废置|流标|失败|作废|异常|撤销|撤回|取消成?交?|流拍|停止)(结果|竞价|项目)?的?(公告|公示|$)|(终止|中止)(采购|招标|询价|议价|竞价|比价|比选|遴选|邀请|邀标|磋商|洽谈|约谈|谈判|拍卖|招租|交易|出让)|关于废置|(招标|询价|询比价?|议价|比选|遴选|磋商)结束的?(公告|公示)|(废标|流标)处理',
  83. '合同公告': '(合同(成交|变更)?)(公告|公示|信息|公式|公开|签订)|合同备案|合同书|合同$', # |(履约|验收)(结果)?
  84. '候选人公示': '候选人(变更)?公示|评标(结果)?(公[告示]|报告)|评审结果|(中标|中选)候选人$|候选人投标文件', #中标前公示|中标预公示|
  85. '中标信息': '(中标|中选|中价|中租|成交)?|入选|确认)(候选人|人|供应商|记录|结果|变更|情况)?的?(公告|公示|结果)|未?入围(公示|公告)|(遴选|采购|招标|竞价|议价|比选|询比?价|评选|谈判|邀标|邀请|洽谈|约谈|评标|发包|磋商|交易|出让|抽取|抽签)\w{,2}结果|单一来源(采购|招标)?的?(中标|成交|结果)|中标通知书|中标$|项目中标|(项目|工程|服务|定点)的?结果公[告示]|超市直购订单', # |开标(记录|信息|情况)
  86. '资审结果': '((资格|资质)(审查|预审|后审|审核)|资审)结果(公告|公示)?|(资质|资格)(预审|后审)公示|资审及业绩公示',
  87. '招标公告': '(采购|招标|询价|询比价?|议价|竞价|比价|比选|遴选|邀请|邀标|磋商|洽谈|约谈|谈判|拍卖|招租|交易|出让)的?(公告|公示|$)|公开(采购|招标|招租|拍卖|挂牌|出让|采招|招采)|(资审|预审|后审)公告',
  88. '开标记录': '开标记录|截标信息|评委名单公示|开标安排|开标数据表|开标信息|开标情况|开标一览表|开标结果|开标会|评审专家公示|开标日程',
  89. '验收合同': '(验收|履约)(公告|公示)|(验收|履约)(结果|报告|意见|单|记录)(公告|公示)|预留项目执行情况|项目验收$'
  90. }
  91. def load_life(self,life_model,config):
  92. with tf.Graph().as_default() as graph:
  93. output_graph_def = graph.as_graph_def()
  94. with open(os.path.join(INTERFACE_DIR, life_model), 'rb') as f:
  95. output_graph_def.ParseFromString(f.read())
  96. tf.import_graph_def(output_graph_def, name='')
  97. # print("%d ops in the final graph" % len(output_graph_def.node))
  98. del output_graph_def
  99. sess = tf.Session(graph=graph,config=config)
  100. sess.run(tf.global_variables_initializer())
  101. inputs = sess.graph.get_tensor_by_name('inputs/inputs:0')
  102. prob = sess.graph.get_tensor_by_name('inputs/dropout:0')
  103. title = sess.graph.get_tensor_by_name('inputs/title:0')
  104. mask = sess.graph.get_tensor_by_name('inputs/mask:0')
  105. mask_title = sess.graph.get_tensor_by_name('inputs/mask_title:0')
  106. # logit = sess.graph.get_tensor_by_name('output/logit:0')
  107. softmax = sess.graph.get_tensor_by_name('output/softmax:0')
  108. return sess, title, inputs, prob, softmax, mask, mask_title
  109. def load_type(self,type_model):
  110. with tf.Graph().as_default() as graph:
  111. output_graph_def = graph.as_graph_def()
  112. with open(os.path.join(INTERFACE_DIR, type_model), 'rb') as f:
  113. output_graph_def.ParseFromString(f.read())
  114. tf.import_graph_def(output_graph_def, name='')
  115. # print("%d ops in the final graph" % len(output_graph_def.node))
  116. del output_graph_def
  117. sess = tf.Session(graph=graph)
  118. sess.run(tf.global_variables_initializer())
  119. inputs = sess.graph.get_tensor_by_name('inputs/inputs:0')
  120. prob = sess.graph.get_tensor_by_name('inputs/dropout:0')
  121. title = sess.graph.get_tensor_by_name('inputs/title:0')
  122. mask = sess.graph.get_tensor_by_name('inputs/mask:0')
  123. mask_title = sess.graph.get_tensor_by_name('inputs/mask_title:0')
  124. # logit = sess.graph.get_tensor_by_name('output/logit:0')
  125. softmax = sess.graph.get_tensor_by_name('output/softmax:0')
  126. return sess, title, inputs, prob, softmax, mask, mask_title
  127. def predict_process(self, docid='', doctitle='', dochtmlcon=''):
  128. # print('准备预处理')
  129. def get_kw_senten(s, span=10):
  130. doc_sens = []
  131. tmp = 0
  132. num = 0
  133. end_idx = 0
  134. for it in re.finditer(self.kws, s): # '|'.join(keywordset)
  135. left = s[end_idx:it.end()].split()
  136. right = s[it.end():].split()
  137. tmp_seg = s[tmp:it.start()].split()
  138. if len(tmp_seg) > span or tmp == 0:
  139. doc_sens.append(' '.join(left[-span:] + right[:span]))
  140. end_idx = it.end() + 1 + len(' '.join(right[:span]))
  141. tmp = it.end()
  142. num += 1
  143. if num >= self.sentence_num:
  144. break
  145. if doc_sens == []:
  146. doc_sens.append(s)
  147. return doc_sens
  148. def word2id(wordlist, max_len=self.sequen_len):
  149. ids = [getIndexOfWords(w) for w in wordlist]
  150. ids = ids[:max_len] if len(ids) >= max_len else ids + [0] * (max_len - len(ids))
  151. assert len(ids) == max_len
  152. return ids
  153. cost_time = dict()
  154. datas = []
  155. datas_title = []
  156. try:
  157. segword_title = ' '.join(selffool.cut(doctitle)[0])
  158. segword_content = dochtmlcon
  159. except:
  160. segword_content = ''
  161. segword_title = ''
  162. if isinstance(segword_content, float):
  163. segword_content = ''
  164. if isinstance(segword_title, float):
  165. segword_title = ''
  166. segword_content = segword_content.replace(' 中 选 ', ' 中选 ').replace(' 中 标 ', ' 中标 ').replace(' 补 遗 ', ' 补遗 '). \
  167. replace(' 更 多', '').replace(' 更多', '').replace(' 中 号 ', ' 中标 ').replace(' 中 选人 ', ' 中选人 '). \
  168. replace(' 点击 下载 查看', '').replace(' 咨询 报价 请 点击', '').replace('终结', '终止')
  169. segword_title = re.sub('[^\s\u4e00-\u9fa5]', '', segword_title)
  170. segword_content = re.sub('[^\s\u4e00-\u9fa5]', '', segword_content)
  171. doc_word_list = segword_content.split()
  172. if len(doc_word_list) > self.sequen_len / 2:
  173. doc_sens = get_kw_senten(' '.join(doc_word_list[100:500]))
  174. doc_sens = ' '.join(doc_word_list[:100]) + '\n' + '\n'.join(doc_sens)
  175. else:
  176. doc_sens = ' '.join(doc_word_list[:self.sequen_len])
  177. # print('标题:',segword_title)
  178. # print('正文:',segword_content)
  179. datas.append(doc_sens.split())
  180. datas_title.append(segword_title.split())
  181. # print('完成预处理')
  182. return datas, datas_title
  183. def is_houxuan(self, title, content):
  184. '''
  185. 通过标题和中文内容判断是否属于候选人公示类别
  186. :param title: 公告标题
  187. :param content: 公告正文文本内容
  188. :return: 1 是候选人公示 ;0 不是
  189. '''
  190. if re.search('候选人的?公示|评标结果|评审结果|中标公示', title): # (中标|成交|中选|入围)
  191. if re.search('变更公告|更正公告|废标|终止|答疑|澄清', title):
  192. return 0
  193. return 1
  194. if re.search('候选人的?公示', content[:100]):
  195. if re.search('公示(期|活动)?已经?结束|公示期已满|中标结果公告|中标结果公示|变更公告|更正公告|废标|终止|答疑|澄清', content[:100]):
  196. return 0
  197. return 1
  198. else:
  199. return 0
  200. def predict(self, title='', list_sentence='', web_source_no='', original_docchannel=''):
  201. not_extract_dic = {
  202. 104: '招标文件',
  203. 106: '法律法规',
  204. 107: '新闻资讯',
  205. 108: '拟建项目',
  206. 109: '展会推广',
  207. 110: '企业名录',
  208. 111: '企业资质',
  209. 112: '全国工程人员',
  210. 113: '业主采购'
  211. }
  212. if original_docchannel in not_extract_dic:
  213. return {'docchannel': {'docchannel':'', 'doctype':not_extract_dic[original_docchannel], "original_docchannel_id": str(original_docchannel)}}
  214. if web_source_no in ['02104-7']:
  215. return {'docchannel': {'docchannel':'', 'doctype':'采招数据'}}
  216. if isinstance(list_sentence, list):
  217. token_l = [it.tokens for it in list_sentence]
  218. tokens = [it for l in token_l for it in l]
  219. content = ' '.join(tokens[:500])
  220. title = re.sub('[^\u4e00-\u9fa5]', '', title)
  221. if len(title)>50:
  222. title = title[:20]+title[-30:]
  223. data_content, data_title = self.predict_process(docid='', doctitle=title[-50:], dochtmlcon=content) # 标题最多取50字
  224. text_len = len(data_content[0]) if len(data_content[0])<self.sequen_len else self.sequen_len
  225. title_len = len(data_title[0]) if len(data_title[0])<self.title_len else self.title_len
  226. result = {'docchannel': {'docchannel':'', 'doctype':'', "original_docchannel_id": str(original_docchannel)}}
  227. array_content = embedding(data_content, shape=(len(data_content), self.sequen_len, 128))
  228. array_title = embedding(data_title, shape=(len(data_title), self.title_len, 128))
  229. pred = self.type_sess.run(self.type_softmax,
  230. feed_dict={
  231. self.type_title: array_title,
  232. self.type_content: array_content,
  233. self.type_mask:[[0]*text_len+[1]*(self.sequen_len-text_len)],
  234. self.type_mask_title:[[0]*title_len+[1]*(self.title_len-title_len)],
  235. self.type_prob:1}
  236. )
  237. id = np.argmax(pred, axis=1)[0]
  238. prob = pred[0][id]
  239. result['docchannel']['doctype'] = self.id2type[id]
  240. # print('公告类别:', self.id2type[id], '概率:',prob)
  241. # if id == 0:
  242. if result['docchannel']['doctype'] not in ['', '新闻资讯']:
  243. pred = self.lift_sess.run(self.lift_softmax,
  244. feed_dict={
  245. self.lift_title: array_title,
  246. self.lift_content: array_content,
  247. self.mask: [[0] * text_len + [1] * (self.sequen_len - text_len)],
  248. self.mask_title: [[0] * title_len + [1] * (self.title_len - title_len)],
  249. self.lift_prob:1}
  250. )
  251. id = np.argmax(pred, axis=1)[0]
  252. prob = pred[0][id]
  253. result['docchannel']['docchannel'] = self.id2life[id]
  254. # print('生命周期:纯模型预测',self.id2life[id], '概率:',prob)
  255. # if id == 6:
  256. if result['docchannel']['docchannel'] == '中标信息':
  257. if self.is_houxuan(''.join([it for it in title if it.isalpha()]), ''.join([it for it in content if it.isalpha()])):
  258. result['docchannel']['docchannel'] = '候选人公示'
  259. # return '候选人公示', prob
  260. # return [{'docchannel': '候选人公示'}]
  261. return result
  262. # return [{'docchannel':self.id2life[id]}]
  263. # else:
  264. # # return self.id2type[id], prob
  265. # return [{'docchannel':self.id2type[id]}]
  266. def predict_rule(self, title, content, channel_dic, prem_dic):
  267. '''2022/2/10加入规则去除某些数据源及内容过短且不包含类别关键词的公告不做预测'''
  268. hetong = '(合同|验收|履约)(公告|公示)|合同号?$' # 合同标题正则
  269. zhongbiao_t = '(中标|中选|成交|入选|入围|结果|确认)(公告|公示|结果)|(遴选|采购|招标|竞价|议价|比选|询价|评选|谈判|邀标|邀请|洽谈|约谈|评标|发包|遴选)结果|开标(记录|信息|情况)|单一来源|直接(选取|选定)|中标通知书|中标$'
  270. zhongbiao_c = '(中标|中选|成交|拟选用|拟邀请|最终选定的?|拟定)(供应商|供货商|服务商|企业|公司|单位|(候选)?人)(名称)?[::]|[,。:.](供应商|供货商|服务商)(名称)?:|指定的中介服务机构:|建设服务单位:'
  271. zhaobiao_t = '(遴选|采购|招标|竞价|议价|比选|询价|评选|谈判|邀标|邀请|洽谈|约谈)(公告|公示|$)'
  272. title_cn = re.sub('[^\u4e00-\u9fa5]', '', title)
  273. if len(re.sub('[^\u4e00-\u9fa5]', "", content))<50 and channel_dic['docchannel']['doctype'] != '新闻资讯':
  274. if re.search(hetong, title_cn) != None:
  275. channel_dic['docchannel']['docchannel'] = '合同公告'
  276. elif re.search(zhongbiao_t, title_cn):
  277. channel_dic['docchannel']['docchannel'] = '中标信息'
  278. elif re.search(zhaobiao_t, title_cn):
  279. channel_dic['docchannel']['docchannel'] = '招标公告'
  280. else:
  281. channel_dic['docchannel']['docchannel'] = ''
  282. elif channel_dic['docchannel'].get('docchannel', '') == '招标公告' and 'win_tenderer' in json.dumps(prem_dic,
  283. ensure_ascii=False):
  284. if re.search(hetong, title_cn) != None:
  285. channel_dic['docchannel']['docchannel'] = '合同公告'
  286. log('正则把招标公告修改为合同公告')
  287. elif re.search(zhongbiao_t, title_cn) or re.search(zhongbiao_t, content[:200]) or re.search(zhongbiao_c,
  288. content):
  289. channel_dic['docchannel']['docchannel'] = '中标信息'
  290. log('正则把招标公告修改为中标信息')
  291. elif channel_dic['docchannel'].get('docchannel', '') == '中标信息' and 'win_tenderer' not in json.dumps(prem_dic,
  292. ensure_ascii=False):
  293. if re.search(hetong, title_cn):
  294. channel_dic['docchannel']['docchannel'] = '合同公告'
  295. log('正则把中标信息修改为合同公告')
  296. elif re.search(zhongbiao_t, title_cn) or re.search(zhongbiao_t, content[:200]) or re.search(zhongbiao_c,
  297. content):
  298. pass
  299. elif re.search(zhaobiao_t, title_cn):
  300. channel_dic['docchannel']['docchannel'] = '招标公告'
  301. log('正则把中标信息修改为招标公告')
  302. elif re.search('中标|成交|中选|入选|入围|结果|供应商|供货商|候选人', title_cn+content)==None:
  303. channel_dic['docchannel']['docchannel'] = ''
  304. log('正则把中标信息修改为空')
  305. return channel_dic
  306. def predict_merge(self, title, list_sentence, html, original_docchannel='', web_source_no=''):
  307. '''
  308. 正则,模型混合预测,返回公告类型及生命周期
  309. :param title: 公告标题
  310. :param content: 预处理后的返回的句子实体列表 list_sentence
  311. :param html: 公告原文 html 内容
  312. :param bidway: 招标方式
  313. :return: {'docchannel': {'docchannel':'中标信息', 'doctype':'采招数据'}} 字典格式
  314. '''
  315. def cut_single_cn_space(text):
  316. new_text = ""
  317. for w in text.split():
  318. if len(w) == 1 or re.search('^[\u4e00-\u9fa5][::]', w):
  319. new_text += w
  320. else:
  321. new_text += ' ' + w
  322. return new_text
  323. def html2text(html):
  324. ser = re.search('<div[^<>]*richTextFetch', html)
  325. # if ser and len(re.sub('[^\u4e00-\u9fa5]', '', html[:ser.start()]))>500:
  326. # html = html[:ser.start()]+'##richTextFetch##'
  327. if ser:
  328. if len(re.sub('[^\u4e00-\u9fa5]', '', html[:ser.start()])) > 200:
  329. html = html[:ser.start()] + '##richTextFetch##'
  330. else:
  331. html = html[:ser.start() + 500]
  332. text = re.sub('<[^<]*?>', '', html).replace('&nbsp;', ' ')
  333. # text = re.sub('http[0-9a-zA-Z-.:/]+|[0-9a-zA-Z-./@]+', '', text)
  334. text = re.sub('\s+', ' ', text)
  335. # text = re.sub('[/|[()()]', '', text)
  336. text = cut_single_cn_space(text)
  337. return text[:20000]
  338. def count_diffser(pattern, text):
  339. num = 0
  340. kw = []
  341. for p in pattern.split(';'):
  342. if re.search(p, text):
  343. num += 1
  344. kw.append(re.search(p, text).group(0))
  345. return num, ';'.join(kw)
  346. def is_single_source(bidway, title):
  347. if re.search('单一来源|单一性采购', title):
  348. return True
  349. elif bidway == '单一来源':
  350. return True
  351. else:
  352. return False
  353. def get_type(title, text):
  354. if re.search(self.title_type_dic['新闻资讯2'], title):
  355. return '新闻资讯', re.search(self.title_type_dic['新闻资讯2'], title).group(0)
  356. if re.search(self.title_type_dic['土地矿产'], title) or re.search(self.type_dic['土地矿产'],text): # and re.search('(土地|用地|宗地|地块)(经营权)?(流转|承包|出租|招租|租赁|确权)', text)==None
  357. if re.search(self.title_type_dic['采招数据'], text.strip().split(' ')[0] + title) \
  358. and not re.search(self.title_type_dic['土地矿产2'], title):
  359. return '采招数据', re.search(self.title_type_dic['采招数据'], text.strip().split(' ')[0] + title).group(0)
  360. return '土地矿产', (re.search(self.title_type_dic['土地矿产'], title) or re.search(self.type_dic['土地矿产'], text)).group(0)
  361. if (re.search(self.title_type_dic['拍卖出让'], title) or re.search(self.type_dic['拍卖出让'], text)):
  362. if re.search(self.title_type_dic['采招数据'], text.strip().split(' ')[0] + title) \
  363. and not re.search(self.title_type_dic['拍卖出让2'], title):
  364. return '采招数据', re.search(self.title_type_dic['采招数据'], text.strip().split(' ')[0] + title).group(0)
  365. return '拍卖出让', (re.search(self.title_type_dic['拍卖出让'], title) or re.search(self.type_dic['拍卖出让'], text)).group(0)
  366. if re.search(self.title_type_dic['产权交易'], title) or re.search(self.type_dic['产权交易'], text):
  367. if re.search(self.title_type_dic['采招数据'], text.strip().split(' ')[0] + title) \
  368. and not re.search(self.title_type_dic['产权交易2'], title):
  369. return '采招数据', re.search(self.title_type_dic['采招数据'], text.strip().split(' ')[0] + title).group(0)
  370. return '产权交易', (re.search(self.title_type_dic['产权交易'], title) or re.search(self.type_dic['产权交易'], text)).group(0)
  371. if re.search(self.title_type_dic['采招数据'], title) or re.search(self.type_dic['采招数据'], title + text):
  372. return '采招数据', (re.search(self.title_type_dic['采招数据'], title) or re.search(self.type_dic['采招数据'], title + text)).group(0)
  373. if (re.search(self.title_type_dic['采招数据'], title +text.strip().split(' ')[0]) or re.search("银行|资格选定|资格认定|资金|公款|存款|存放|现金管理|招募|入围|入库", title +text.strip().split(' ')[0])) and \
  374. not re.search(self.title_type_dic['新闻资讯2'], title):
  375. return '采招数据', (re.search(self.title_type_dic['采招数据'], title +text.strip().split(' ')[0]) or re.search("银行|资格选定|资格认定|资金|公款|存款|存放|现金管理|招募|入围|入库", title +text.strip().split(' ')[0])).group(0)
  376. if re.search(self.title_type_dic['新闻资讯'], title):
  377. return '新闻资讯', re.search(self.title_type_dic['新闻资讯'], title).group(0)
  378. # else:
  379. return '', '没有公告类型关键词,返回空'
  380. def get_life(title, text ,doc_type=None):
  381. def words_dump(list_words):
  382. _words_l_l = []
  383. list_words.sort(key=lambda x: len(x))
  384. for _l in list_words:
  385. _exists = False
  386. for l1 in _words_l_l:
  387. if l1 in _l:
  388. _exists = True
  389. break
  390. if not _exists:
  391. _words_l_l.append(_l)
  392. return _words_l_l
  393. title = re.sub('[-()()0-9a-z]|第?[二三四]次公?告?', '', title)
  394. first_line = text.split()[0] if len(text.split()) > 2 else ''
  395. if title.strip()[-2:] not in ['公告', '公示'] and 5 < len(first_line) < 50 and first_line[-2:] in ['公告', '公示']:
  396. # print('title: ', title, first_line)
  397. title += first_line
  398. def count_score(l):
  399. return len(l) + len(set(l)) * 2
  400. life_kw_title = {}
  401. life_kw_content = {}
  402. life_score = {}
  403. # msc = ""
  404. # 查找标题每个类别关键词
  405. for k, v in self.title_life_dic.items():
  406. k2 = re.sub('[\da-z]', '', k)
  407. if k2 not in life_kw_title:
  408. life_kw_title[k2] = []
  409. for it in re.finditer(v, title):
  410. life_kw_title[k2].append(it.group(0))
  411. # 查找正文每个类别关键词
  412. for k, v in self.life_dic.items():
  413. k2 = re.sub('[\da-z]', '', k)
  414. if k2 not in life_kw_content:
  415. life_kw_content[k2] = {'pos': [], 'neg': []}
  416. for it in re.finditer(v, text):
  417. if 'neg' not in k:
  418. life_kw_content[k2]['pos'].append(it.group(0))
  419. else:
  420. life_kw_content[k2]['neg'].append(it.group(0))
  421. for k2 in life_kw_content:
  422. life_kw_content[k2]['pos'] = words_dump(life_kw_content[k2]['pos'])# 关键词去重
  423. life_kw_content[k2]['neg'] = words_dump(life_kw_content[k2]['neg'])
  424. life_score[k2] = count_score(life_kw_content[k2]['pos']) - count_score(
  425. life_kw_content[k2]['neg'])
  426. life_kw_title = {k: v for k, v in life_kw_title.items() if v != []}
  427. life_kw_content = {k: v for k, v in life_kw_content.items() if life_score[k] > 0}
  428. msc = [life_kw_title, life_kw_content, life_score]
  429. msc = json.dumps(msc, ensure_ascii=False)
  430. max_score = 0
  431. life_list = []
  432. for k in life_score.keys():
  433. if life_score[k] > max_score:
  434. max_score = life_score[k]
  435. life_list = [k]
  436. elif life_score[k] == max_score and life_score[k] > 0:
  437. life_list.append(k)
  438. #add 关键词规则明确时可直接返回,符合多个规则时调用新模型
  439. if doc_type=="采招数据" and len(text)>300:
  440. life_list_in_title = [l for l in life_list if l in life_kw_title]# 标题正文同时有关键词的
  441. if len(life_list_in_title)==1:
  442. life_list = life_list_in_title
  443. life_kw_title = {life_list_in_title[0]:life_kw_title[life_list_in_title[0]]}
  444. # elif len(life_list_in_title)==0:
  445. # return "", msc
  446. if len(life_kw_title)==0:
  447. if len(life_list)==0 or len(life_list)>1:# 关键词最高score有多个相同,直接用新模型
  448. return "", msc
  449. if '采购意向' in life_kw_title or '采购意向' in life_list:
  450. if '中标信息' in life_kw_title or '中标信息' in life_list:
  451. return '中标信息', msc
  452. elif '候选人公示' in life_kw_title:
  453. return '候选人公示', msc
  454. elif set(['候选人公示', '合同公告']) & set(life_kw_title) != set():
  455. return '', msc
  456. return '采购意向', msc
  457. elif '招标预告' in life_kw_title or '招标预告' in life_list:
  458. if '中标信息' in life_kw_title or '中标信息' in life_list:
  459. return '中标信息', msc
  460. elif '候选人公示' in life_kw_title:
  461. return '候选人公示', msc
  462. elif set(['候选人公示', '合同公告']) & set(life_kw_title) != set():
  463. return '', msc
  464. return '招标预告', msc
  465. elif '公告变更' in life_kw_title or '公告变更' in life_list:
  466. if life_score.get('候选人公示', 0) > 3 or '候选人公示' in life_kw_title:
  467. return '候选人公示', msc
  468. elif life_score.get('合同公告', 0) > 3 or '合同公告' in life_kw_title:
  469. return '合同公告', msc
  470. elif life_score.get('中标信息', 0) > 3 or '中标信息' in life_kw_title:
  471. return '中标信息', msc
  472. elif '招标公告' in life_kw_title and re.search('变更|更正|更改|延期', title[-4:])==None and life_score.get('公告变更', 0) < 4:
  473. return '招标公告', msc
  474. return '公告变更', msc
  475. elif '招标答疑' in life_kw_title or '招标答疑' in life_list:
  476. if '招标公告' in life_kw_title and life_score.get('招标答疑', 0) < 4:
  477. return '招标公告', msc
  478. elif life_score.get('招标答疑', 0) < max_score:
  479. if max_score > 3 and len(life_list) == 1:
  480. return life_list[0], msc
  481. return '', msc
  482. return '招标答疑', msc
  483. elif '开标记录' in life_kw_title:
  484. return '开标记录', msc
  485. elif '验收合同' in life_kw_title:
  486. return '验收合同', msc
  487. elif '候选人公示' in life_kw_title or '候选人公示' in life_list:
  488. if '招标公告' in life_kw_title and '候选人公示' not in life_kw_title: # and life_score.get('招标公告', 0) > 3
  489. return '招标公告', msc
  490. elif '废标公告' in life_kw_title or life_score.get('废标公告', 0) > 5:
  491. return '废标公告', msc
  492. return '候选人公示', msc
  493. elif '合同公告' in life_kw_title or '合同公告' in life_list:
  494. if '招标公告' in life_kw_title and life_score.get('招标公告', 0) > 3:
  495. return '招标公告', msc
  496. elif '废标公告' in life_kw_title or life_score.get('废标公告', 0) > 5:
  497. return '废标公告', msc
  498. return '合同公告', msc
  499. elif '中标信息' in life_kw_title or '中标信息' in life_list:
  500. if '招标公告' in life_kw_title and '中标信息' not in life_kw_title and life_score.get('招标公告',0) >= life_score.get('中标信息',0): # (life_score.get('招标公告', 0)>2 or life_score.get('中标信息', 0)<4) 0.7886409793924245
  501. return '招标公告', msc
  502. elif '废标公告' in life_kw_title or life_score.get('废标公告', 0) > 5:
  503. return '废标公告', msc
  504. elif life_score.get('候选人公示', 0) > 3:
  505. return '候选人公示', msc
  506. elif life_score.get('合同公告', 0) > 5:
  507. return '合同公告', msc
  508. return '中标信息', msc
  509. elif '废标公告' in life_kw_title or '废标公告' in life_list:
  510. if life_score.get('招标公告', 0) > 3 and '废标公告' not in life_kw_title:
  511. return '招标公告', msc
  512. return '废标公告', msc
  513. elif '资审结果' in life_kw_title or '资审结果' in life_list:
  514. return '资审结果', msc
  515. elif '招标公告' in life_kw_title or '招标公告' in life_list:
  516. return '招标公告', msc
  517. return '', msc
  518. def get_model_inputs(list_sentence):
  519. list_sentence = sorted(list_sentence, key=lambda x:x.sentence_index)
  520. token_l = [it.tokens for it in list_sentence]
  521. tokens = [it for l in token_l for it in l]
  522. content = ' '.join(tokens[:500])
  523. data_content, data_title = self.predict_process(docid='', doctitle=title[-50:],
  524. dochtmlcon=content) # 标题最多取50字
  525. text_len = len(data_content[0]) if len(data_content[0]) < self.sequen_len else self.sequen_len
  526. title_len = len(data_title[0]) if len(data_title[0]) < self.title_len else self.title_len
  527. array_content = embedding(data_content, shape=(len(data_content), self.sequen_len, 128))
  528. array_title = embedding(data_title, shape=(len(data_title), self.title_len, 128))
  529. return array_content, array_title ,text_len, title_len, content
  530. def type_model_predict():
  531. pred = self.type_sess.run(self.type_softmax,
  532. feed_dict={
  533. self.type_title: array_title,
  534. self.type_content: array_content,
  535. self.type_mask: [[0] * text_len + [1] * (self.sequen_len - text_len)],
  536. self.type_mask_title: [[0] * title_len + [1] * (self.title_len - title_len)],
  537. self.type_prob: 1}
  538. )
  539. id = np.argmax(pred, axis=1)[0]
  540. prob = pred[0][id]
  541. return id, prob
  542. def life_model_predict():
  543. pred = self.lift_sess.run(self.lift_softmax,
  544. feed_dict={
  545. self.lift_title: array_title,
  546. self.lift_content: array_content,
  547. self.mask: [[0] * text_len + [1] * (self.sequen_len - text_len)],
  548. self.mask_title: [[0] * title_len + [1] * (self.title_len - title_len)],
  549. self.lift_prob: 1}
  550. )
  551. id = np.argmax(pred, axis=1)[0]
  552. prob = pred[0][id]
  553. return id, prob
  554. not_extract_dic = {
  555. 104: '招标文件',
  556. 106: '法律法规',
  557. 107: '新闻资讯',
  558. 108: '拟建项目',
  559. 109: '展会推广',
  560. 110: '企业名录',
  561. 111: '企业资质',
  562. 112: '全国工程人员',
  563. 113: '业主采购'
  564. }
  565. origin_dic = {51: '公告变更',
  566. 52: '招标公告',
  567. 101: '中标信息',
  568. 102: '招标预告',
  569. 103: '招标答疑',
  570. 104: '招标文件',
  571. 105: '资审结果',
  572. 106: '法律法规',
  573. 107: '新闻资讯',
  574. 108: '拟建项目',
  575. 109: '展会推广',
  576. 110: '企业名录',
  577. 111: '企业资质',
  578. 112: '全国工程',
  579. 113: '业主采购',
  580. 114: '采购意向',
  581. 115: '拍卖出让',
  582. 116: '土地矿产',
  583. 117: '产权交易',
  584. 118: '废标公告',
  585. 119: '候选人公示',
  586. 120: '合同公告'}
  587. title = re.sub('[^\u4e00-\u9fa5《》]+|出租车', '', title)
  588. if len(title) > 50:
  589. title = title[:20] + title[-30:]
  590. text = html2text(html)
  591. self.origin_dic = origin_dic
  592. self.title = title
  593. self.text = text
  594. if original_docchannel in not_extract_dic:
  595. return {'docchannel': {'docchannel': '', 'doctype': not_extract_dic[original_docchannel], 'life_docchannel': origin_dic.get(original_docchannel, '原始类别')}}, '公告类别不在提取范围'
  596. if web_source_no in ['02104-7', '04733', 'DX007628-6']: # 这些数据源无法识别
  597. return {'docchannel': {'docchannel': origin_dic.get(original_docchannel, '原始类别'), 'doctype': '采招数据', 'life_docchannel': origin_dic.get(original_docchannel, '原始类别')}}, '此数据源公告分类不明确,返回数据源类别'
  598. if original_docchannel == 303:
  599. return {'docchannel': {'docchannel': '处罚公告', 'doctype': '处罚公告', 'life_docchannel': '处罚公告'}}, "源类别为处罚公告"
  600. result = {'docchannel': {'docchannel': '', 'doctype': ''}}
  601. doc_type, type_kw = get_type(title, text)
  602. # print(doc_type, type_kw)
  603. # doc_life, life_kw = get_life(title, text, prem_json, bidway, original_docchannel)
  604. # doc_life, life_kw = get_life(title, text)
  605. doc_life, life_kw = get_life(title, text[:5000],doc_type=doc_type)
  606. # print('doc_life',doc_life, life_kw)
  607. if doc_type in self.title_type_dic:
  608. result['docchannel']['doctype'] = doc_type
  609. if doc_life in self.title_life_dic:
  610. result['docchannel']['docchannel'] = doc_life
  611. # print('channel正则预测结果:', result)
  612. msc = '正则结果:类型:%s, 关键词:%s, 周期:%s, 关键词:%s'%(doc_type, type_kw,doc_life, life_kw)+'\n'+'模型结果:'
  613. # print('类型:%s, 关键词:%s, 周期:%s, 关键词:%s'%(doc_type, type_kw,doc_life, life_kw))
  614. if doc_type == "" or doc_life == "" or (doc_type != '采招数据' and origin_dic.get(original_docchannel, '原始类别') in ['招标公告', '中标信息', '招标预告', '采购意向']):
  615. array_content, array_title, text_len, title_len, content = get_model_inputs(list_sentence)
  616. if doc_type == "" or (doc_type != '采招数据' and origin_dic.get(original_docchannel, '原始类别') in ['招标公告', '中标信息', '招标预告', '采购意向']) and \
  617. not re.search(self.title_type_dic["新闻资讯2"],title):
  618. type_id, type_prob = type_model_predict()
  619. type_model = self.id2type[type_id]
  620. if type_model == '新闻资讯' and doc_life!='': # 修复bug 78584245 "docchannel": "合同公告", "doctype": "新闻资讯",
  621. result['docchannel']['doctype'] = '采招数据'
  622. msc += '模型结果为新闻资讯,生命周期不为空,改为采招数据;'
  623. else:
  624. result['docchannel']['doctype'] = type_model
  625. msc += type_model + ' 概率:%.4f;'%type_prob
  626. # print('公告类别:', self.id2type[id], '概率:',prob)
  627. if result['docchannel']['doctype'] in ('产权交易', '土地矿产', '拍卖出让'):
  628. # 竞买类公告,取最低价成交的修正为‘采招数据’
  629. if re.search("竞买",title) and re.search("最低价.{,2}成交|中标为最低报价|出价最低者为中标",text):
  630. result['docchannel']['doctype'] = '采招数据'
  631. # doc_type为采招数据时重新预测doc_life
  632. if result['docchannel']['doctype'] == "采招数据":
  633. doc_life, life_kw = get_life(title, text[:5000], doc_type="采招数据")
  634. if doc_life=="" and result['docchannel']['doctype'] not in ['', '新闻资讯']:
  635. if result['docchannel']['doctype'] != "采招数据" or len(text)<=300: # doc_type为采招数据或公告过短时,使用新模型预测
  636. # if len(text)>150 and re.search(self.kws, content):
  637. if len(text)>100 and re.search(self.kws, content):
  638. life_id, life_prob = life_model_predict()
  639. if life_prob>=0.8 or (len(text)<=300 and life_prob>=0.6):
  640. life_model = self.id2life[life_id]
  641. result['docchannel']['docchannel'] = life_model
  642. msc += life_model + ' 概率:%.4f;\n'%life_prob
  643. # msc = final_change(msc)
  644. # print('channel ', msc)
  645. return result, msc
  646. def final_change(self, result, prem, original_docchannel,nlp_enterprise,nlp_enterprise_attachment, msc):
  647. '''
  648. :param result: channel 结果字典
  649. :param prem:
  650. :param title: 标题
  651. :param text: 正文
  652. :param original_docchannel: 站源类别
  653. :param msc: 备注
  654. :return: channel结果字典
  655. '''
  656. '''
  657. 修改逻辑:
  658. 1、中标公告、合同公告无中标人且原始为非中标,返回原类型
  659. 2、废标公告有中标人且标题无废标关键词,返回中标信息
  660. 3、答疑公告标题无答疑关键且原始为招标,返回原始类别
  661. 4、招标公告有中标人且原始为中标,返回中标信息
  662. 5、预测为招标,原始为预告、意向,返回原始类别
  663. 6、预测及原始均在变更、答疑,返回原始类别
  664. 7、预测为采招数据,原始为产权且有关键词,返回原始类别
  665. 8、废标公告原始为招标、预告且标题无废标关键期,返回原始类别
  666. 9、若预测为非采招数据且源网为采招数据且有招标关键词返回采招数据
  667. 10、招标公告有中标人,且标题有直购关键词,改为中标信息
  668. 11、预测预告,原始为意向、招标且标题无预告关键词,返回原始类别
  669. '''
  670. def is_contain_winner(extract_json):
  671. if re.search('win_tenderer', extract_json):
  672. return True
  673. else:
  674. return False
  675. tenderee = ""
  676. agency = ""
  677. try:
  678. for k, v in prem['prem'].items():
  679. for link in v['roleList']:
  680. if link['role_name'] == 'tenderee' and tenderee == "":
  681. tenderee = link['role_text']
  682. if link['role_name'] == 'agency' and agency == "":
  683. agency = link['role_text']
  684. except Exception as e:
  685. # print('解析prem 获取招标人、代理人出错')
  686. pass
  687. origin_dic = self.origin_dic
  688. title = self.title
  689. text = self.text
  690. # 剔除招标单位名称影响
  691. if tenderee:
  692. title = title.replace(tenderee, " ")
  693. text = text.replace(tenderee, " ")
  694. if agency:
  695. title = title.replace(agency, " ")
  696. text = text.replace(agency, " ")
  697. enterprise_list = list(set(nlp_enterprise + nlp_enterprise_attachment))
  698. enterprise_list.sort(key = lambda x:len(x), reverse=True)
  699. for enterprise in enterprise_list:
  700. if enterprise and enterprise not in (tenderee,agency):
  701. text = text.replace(enterprise, " ")
  702. prem_json = json.dumps(prem, ensure_ascii=False)
  703. if result['docchannel']['docchannel'] in ['中标信息', '合同公告'] and origin_dic.get(
  704. original_docchannel, '') in ['招标公告', '采购意向', '招标预告', '公告变更'] and is_contain_winner(
  705. prem_json) == False and re.search(self.title_life_dic['中标信息'], title+text[:40]) == None:
  706. result['docchannel']['docchannel'] = origin_dic.get(original_docchannel, '')
  707. msc += '最终规则修改:中标公告、合同公告无中标人且原始为非中标,返回原类型'
  708. elif result['docchannel']['docchannel'] in ['中标信息'] and is_contain_winner(prem_json) == False \
  709. and re.search("监督(抽查|检查)结果|抽查结果", title):
  710. result['docchannel']['doctype'] = "新闻资讯"
  711. result['docchannel']['docchannel'] = ""
  712. msc += '最终规则修改:中标公告无中标人且包含新闻资讯关键词,返回新闻资讯类型'
  713. elif result['docchannel']['docchannel'] in ['中标信息'] and is_contain_winner(prem_json) == False \
  714. and (re.search(self.title_life_dic['废标公告'],title) or re.search(self.life_dic['废标公告'],text[:400]) or re.search(self.life_dic['废标公告2'],text[:400])):
  715. result['docchannel']['docchannel'] = "废标公告"
  716. msc += '最终规则修改:中标公告无中标人且包含废标公告关键词,改为废标公告'
  717. elif result['docchannel']['docchannel'] == '废标公告' and is_contain_winner(prem_json) and re.search(
  718. self.title_life_dic['废标公告'], title) == None:
  719. if re.search(self.title_life_dic['合同公告'], title):
  720. result['docchannel']['docchannel'] = '合同公告'
  721. else:
  722. result['docchannel']['docchannel'] = '中标信息'
  723. msc += '最终规则修改:预测为废标却有中标人且标题无废标关键词改为中标信息;'
  724. elif result['docchannel']['docchannel'] in ['招标答疑'] and re.search(
  725. self.title_life_dic['招标答疑'], title) == None and origin_dic.get(
  726. original_docchannel, '') in ['招标公告', '采购意向', '招标预告']:
  727. result['docchannel']['docchannel'] = origin_dic.get(original_docchannel, '')
  728. msc += '最终规则修改:答疑公告标题无答疑关键且原始为招标,返回原始类别;'
  729. elif result['docchannel']['docchannel'] == '招标公告' and is_contain_winner(prem_json) and (origin_dic.get(
  730. original_docchannel, '') == '中标信息' or re.search('直接采购|单源直采|直采', title)): # 20241025补充 标题包含直接采购且有中标人的为中标公告
  731. result['docchannel']['docchannel'] = '中标信息'
  732. msc += '最终规则修改:预测为招标公告却有中标人且原始为中标改为中标信息;'
  733. elif result['docchannel']['docchannel'] in ['招标公告'] and origin_dic.get(
  734. original_docchannel, '') in ['采购意向', '招标预告']:
  735. result['docchannel']['docchannel'] = origin_dic.get(original_docchannel, '')
  736. msc += '最终规则修改:预测为招标,原始为预告、意向,返回原始类别'
  737. elif result['docchannel']['docchannel'] in ['招标预告'] and origin_dic.get(
  738. original_docchannel, '') in ['采购意向', '招标公告'] and re.search(
  739. self.title_life_dic['招标预告'], title) == None and re.search(
  740. '预公?告|预公示|报建公告|(批前|标前)公示|(供应|招标|采购)计划表?|(论证|征求|征集)(供应商)?意见|意见征询|需求评审公告|需求(公告|公示|意见)',text[:40]) == None:
  741. result['docchannel']['docchannel'] = origin_dic.get(original_docchannel, '')
  742. msc += '最终规则修改:预测预告,原始为意向、招标且标题无预告关键词,返回原始类别'
  743. elif result['docchannel']['docchannel'] in ['招标答疑', '公告变更'] and origin_dic.get(
  744. original_docchannel, '') in ['招标答疑', '公告变更']:
  745. if re.search(self.title_life_dic['公告变更'], title):
  746. result['docchannel']['docchannel'] = "公告变更"
  747. msc += '最终规则修改:预测及原始均在答疑、变更,返回原始类别'
  748. elif re.search(self.title_life_dic['招标答疑'], title):
  749. result['docchannel']['docchannel'] = "招标答疑"
  750. msc += '最终规则修改:预测及原始均在答疑、变更,返回原始类别'
  751. else:
  752. result['docchannel']['docchannel'] = origin_dic.get(original_docchannel, '')
  753. msc += '最终规则修改:预测及原始均在答疑、变更,返回原始类别'
  754. elif result['docchannel']['doctype'] == '采招数据' and origin_dic.get(
  755. original_docchannel, '') in ['产权交易', '土地矿产'] and re.search('产权|转让|受让|招租|出租|承租|竞价', text) and not re.search('公开(采购|招标|采招|招采)', title):
  756. result['docchannel']['doctype'] = origin_dic.get(original_docchannel, '')
  757. # print(re.findall('产权|转让|受让|招租|出租|承租|竞价', text))
  758. msc += '最终规则修改:预测为采招数据,原始为产权且有关键词,返回原始类别'
  759. elif result['docchannel']['docchannel'] == '废标公告' and origin_dic.get(
  760. original_docchannel, '') in ['招标公告', '采购意向', '招标预告'] and re.search(
  761. self.title_life_dic['废标公告'], title+text[:50]) == None and re.search(
  762. self.life_dic['废标公告'], title+text[:50]) == None and re.search(
  763. self.life_dic['废标公告2'], title+text[:50]) == None and re.search(
  764. self.title_life_dic['中标信息'], title) == None:
  765. result['docchannel']['docchannel'] = origin_dic.get(original_docchannel, '')
  766. msc += '最终规则修改:废标公告原始为招标、预告且标题无废标关键词,返回原始类别;'
  767. elif result['docchannel']['docchannel'] in ['招标公告', '招标预告'] and is_contain_winner(
  768. prem_json) and re.search('直购', title):
  769. result['docchannel']['docchannel'] = '中标信息'
  770. msc += "最终规则修改:预测为招标却有中标人且标题有直购关键词返回中标"
  771. elif result['docchannel']['docchannel'] == '开标记录' and '开标结果' in title and is_contain_winner(prem_json):
  772. msc += "最终规则修改:开标结果包含中标人的作为中标信息"
  773. result['docchannel']['docchannel'] = '中标信息'
  774. if result['docchannel']['doctype'] in ['产权交易', '土地矿产', '拍卖出让'] and origin_dic.get(
  775. original_docchannel, '') not in ['产权交易', '土地矿产', '拍卖出让'] \
  776. and (re.search(self.title_type_dic['采招数据'], title) or re.search('采购|询价|磋商', title)
  777. or re.search('(采购|招投?标|投标)(信息|内容|项目|公告|数量|人|单位|方式)|(建设|工程|服务|施工|监理|勘察|设计)项目|(%s)'
  778. % self.type_dic['采招数据'], text)
  779. ):
  780. # print('test',re.findall('(采购|招投?标|投标)(信息|内容|项目|公告|数量|人|单位|方式)|(建设|工程|服务|施工|监理|勘察|设计)项目|(%s)' % self.type_dic['采招数据'], text))
  781. result['docchannel']['doctype'] = '采招数据'
  782. msc += ' 最终规则修改:预测为非采招数据,原始为采招数据且有招标关键词,返回采招数据'
  783. elif result['docchannel']['doctype'] in ['土地矿产'] and origin_dic.get(original_docchannel, '') in ['拍卖出让', '产权交易']:
  784. if origin_dic.get(original_docchannel, '') in ['拍卖出让'] and (
  785. re.search(self.title_type_dic['拍卖出让'], title) or re.search(self.type_dic['拍卖出让'], text)):
  786. result['docchannel']['doctype'] = '拍卖出让'
  787. msc += "最终规则修改:预测为土地矿产原始为拍卖且有拍卖关键词,返回拍卖"
  788. elif (re.search(self.title_type_dic['产权交易'], title) or re.search(self.type_dic['产权交易'], text)):
  789. result['docchannel']['doctype'] = '产权交易'
  790. msc += "最终规则修改:预测为土地矿产原始为产权交易且有产权交易关键词,返回产权交易"
  791. '''下面是新格式增加返回字段'''
  792. if result['docchannel']['docchannel'] != '': # 预测到生命周期的复制到life_docchannel,否则用数据源结果
  793. result['docchannel']['life_docchannel'] = result['docchannel']['docchannel']
  794. else:
  795. result['docchannel']['life_docchannel'] = origin_dic.get(original_docchannel, '原始类别')
  796. return result, msc