|
|
@@ -25,6 +25,7 @@ from bs4 import BeautifulSoup
|
|
|
import copy
|
|
|
import calendar
|
|
|
import datetime
|
|
|
+import fool
|
|
|
|
|
|
from threading import RLock
|
|
|
dict_predictor = {"codeName":{"predictor":None,"Lock":RLock()},
|
|
|
@@ -40,7 +41,8 @@ dict_predictor = {"codeName":{"predictor":None,"Lock":RLock()},
|
|
|
"product_attrs":{"predictor":None,"Lock":RLock()},
|
|
|
"channel": {"predictor": None, "Lock": RLock()},
|
|
|
"deposit_payment_way": {"predictor": None, "Lock": RLock()},
|
|
|
- "total_unit_money": {"predictor": None, "Lock": RLock()}
|
|
|
+ "total_unit_money": {"predictor": None, "Lock": RLock()},
|
|
|
+ "industry": {"predictor": None, "Lock": RLock()}
|
|
|
}
|
|
|
|
|
|
|
|
|
@@ -76,6 +78,8 @@ def getPredictor(_type):
|
|
|
dict_predictor[_type]["predictor"] = DepositPaymentWay()
|
|
|
if _type == 'total_unit_money':
|
|
|
dict_predictor[_type]["predictor"] = TotalUnitMoney()
|
|
|
+ if _type == 'industry':
|
|
|
+ dict_predictor[_type]["predictor"] = IndustryPredictor()
|
|
|
return dict_predictor[_type]["predictor"]
|
|
|
raise NameError("no this type of predictor")
|
|
|
|
|
|
@@ -1818,6 +1822,7 @@ class ProductPredictor():
|
|
|
with self.sess.as_default() as sess:
|
|
|
with self.sess.graph.as_default():
|
|
|
result = []
|
|
|
+ product_list = []
|
|
|
if fail and list_articles!=[]:
|
|
|
text_list = [list_articles[0].content[:MAX_AREA]]
|
|
|
chars = [[self.word2index.get(it, self.word2index.get('<unk>')) for it in text] for text in text_list]
|
|
|
@@ -1835,6 +1840,7 @@ class ProductPredictor():
|
|
|
batch_paths = self.decode(scores, lengths, tran_)
|
|
|
for text, path, length in zip(text_list, batch_paths, lengths):
|
|
|
tags = ''.join([str(it) for it in path[:length]])
|
|
|
+ # 提取产品
|
|
|
for it in re.finditer("12*3", tags):
|
|
|
start = it.start()
|
|
|
end = it.end()
|
|
|
@@ -1845,6 +1851,8 @@ class ProductPredictor():
|
|
|
begin_index=0, end_index=0, wordOffset_begin=start,
|
|
|
wordOffset_end=end)
|
|
|
list_entitys[0].append(_entity)
|
|
|
+ product_list.append(text[start:end])
|
|
|
+ # 提取失败原因
|
|
|
for it in re.finditer("45*6", tags):
|
|
|
start = it.start()
|
|
|
end = it.end()
|
|
|
@@ -1858,7 +1866,7 @@ class ProductPredictor():
|
|
|
reasons.append(it)
|
|
|
elif reasons == []:
|
|
|
reasons.append(it)
|
|
|
- return {'fail_reason':'；'.join(reasons)}
|
|
|
+ return {'fail_reason':'；'.join(reasons)}, product_list
|
|
|
|
|
|
if list_entitys is None:
|
|
|
list_entitys = [[] for _ in range(len(list_sentences))]
|
|
|
@@ -1903,6 +1911,8 @@ class ProductPredictor():
|
|
|
wordOffset_end=end,in_attachment=sentence.in_attachment)
|
|
|
list_entity.append(_entity)
|
|
|
temp_list.append(sentence.sentence_text[start:end])
|
|
|
+ product_list.append(sentence.sentence_text[start:end])
|
|
|
+
|
|
|
# item["product"] = list(set(temp_list))
|
|
|
# result.append(item)
|
|
|
if _begin_index+_LEN >= len(list_sentence):
|
|
|
@@ -1910,7 +1920,7 @@ class ProductPredictor():
|
|
|
_begin_index += _LEN
|
|
|
item["product"] = list(set(temp_list))
|
|
|
result.append(item) # 修正bug
|
|
|
- return {'fail_reason': ""}
|
|
|
+ return {'fail_reason': ""},product_list
|
|
|
|
|
|
|
|
|
# 产品数量单价品牌规格提取 #2021/11/10 添加表格中的项目、需求、预算、时间要素提取
|
|
|
@@ -3199,6 +3209,537 @@ class TotalUnitMoney:
|
|
|
# print("total_unit_money", _entity.entity_text,
|
|
|
# _entity.is_total_money, _entity.is_unit_money)
|
|
|
|
|
|
+# 行业分类
|
|
|
+class IndustryPredictor():
|
|
|
+ def __init__(self,):
|
|
|
+ self.model_path = os.path.dirname(__file__)+ '/industry_model'
|
|
|
+ self.id2lb = {0: '专业施工', 1: '专用仪器仪表', 2: '专用设备修理', 3: '互联网信息服务', 4: '互联网安全服务', 5: '互联网平台', 6: '互联网接入及相关服务', 7: '人力资源服务',
|
|
|
+ 8: '人造原油', 9: '仓储业', 10: '仪器仪表', 11: '仪器仪表修理', 12: '会计、审计及税务服务', 13: '会议、展览及相关服务', 14: '住宅、商业用房',
|
|
|
+ 15: '体育场地设施管理', 16: '体育组织', 17: '体育设备', 18: '保险服务', 19: '信息处理和存储支持服务', 20: '信息技术咨询服务',
|
|
|
+ 21: '信息系统集成和物联网技术服务', 22: '修缮工程', 23: '健康咨询', 24: '公路旅客运输', 25: '其他专业咨询与调查', 26: '其他专业技术服务',
|
|
|
+ 27: '其他交通运输设备', 28: '其他公共设施管理', 29: '其他土木工程建筑', 30: '其他工程服务', 31: '其他建筑建材', 32: '其他运输业', 33: '农业和林业机械',
|
|
|
+ 34: '农业服务', 35: '农产品', 36: '农副食品，动、植物油制品', 37: '出版业', 38: '办公消耗用品及类似物品', 39: '办公设备', 40: '化学原料及化学制品',
|
|
|
+ 41: '化学纤维', 42: '化学药品和中药专用设备', 43: '医疗设备', 44: '医药品', 45: '卫星传输服务', 46: '卫生', 47: '印刷服务', 48: '图书和档案',
|
|
|
+ 49: '图书档案设备', 50: '图书馆与档案馆', 51: '土地管理业', 52: '地质勘查', 53: '地震服务', 54: '场馆、站港用房', 55: '城市公共交通运输',
|
|
|
+ 56: '塑料制品、半成品及辅料', 57: '天然石料', 58: '娱乐设备', 59: '婚姻服务', 60: '安全保护服务', 61: '安全生产设备', 62: '家具用具',
|
|
|
+ 63: '家用电器修理', 64: '工业、生产用房', 65: '工业与专业设计及其他专业技术服务', 66: '工矿工程建筑', 67: '工程技术与设计服务', 68: '工程机械',
|
|
|
+ 69: '工程监理服务', 70: '工程评价服务', 71: '工程造价服务', 72: '市场调查', 73: '广告业', 74: '广播', 75: '广播、电视、电影设备',
|
|
|
+ 76: '广播电视传输服务', 77: '废弃资源综合利用业', 78: '建筑涂料', 79: '建筑物、构筑物附属结构', 80: '建筑物拆除和场地准备活动', 81: '建筑装饰和装修业',
|
|
|
+ 82: '录音制作', 83: '影视节目制作', 84: '房地产中介服务', 85: '房地产开发经营', 86: '房地产租赁经营', 87: '房屋租赁', 88: '招标代理',
|
|
|
+ 89: '探矿、采矿、选矿和造块设备', 90: '政法、检测专用设备', 91: '教育服务', 92: '教育设备', 93: '文物及非物质文化遗产保护', 94: '文物和陈列品',
|
|
|
+ 95: '文艺创作与表演', 96: '文艺设备', 97: '新闻业', 98: '旅行社及相关服务', 99: '日杂用品', 100: '有色金属冶炼及压延产品', 101: '有色金属矿',
|
|
|
+ 102: '木材、板材等', 103: '木材采集和加工设备', 104: '机械设备', 105: '机械设备经营租赁', 106: '林业产品', 107: '林业服务', 108: '架线和管道工程建筑',
|
|
|
+ 109: '核工业专用设备', 110: '橡胶制品', 111: '殡葬服务', 112: '殡葬设备及用品', 113: '气象服务', 114: '水上交通运输设备', 115: '水上运输业',
|
|
|
+ 116: '水利和水运工程建筑', 117: '水工机械', 118: '水文服务', 119: '水资源管理', 120: '污水处理及其再生利用', 121: '汽车、摩托车修理与维护',
|
|
|
+ 122: '法律服务', 123: '洗染服务', 124: '测绘地理信息服务', 125: '海洋仪器设备', 126: '海洋工程建筑', 127: '海洋服务', 128: '消防设备',
|
|
|
+ 129: '清洁服务', 130: '渔业产品', 131: '渔业服务', 132: '炼焦和金属冶炼轧制设备', 133: '烟草加工设备', 134: '热力生产和供应', 135: '焦炭及其副产品',
|
|
|
+ 136: '煤炭采选产品', 137: '燃气生产和供应业', 138: '物业管理', 139: '特种用途动、植物', 140: '环保咨询', 141: '环境与生态监测检测服务',
|
|
|
+ 142: '环境污染防治设备', 143: '环境治理业', 144: '玻璃及其制品', 145: '理发及美容服务', 146: '生态保护', 147: '电信',
|
|
|
+ 148: '电力、城市燃气、蒸汽和热水、水', 149: '电力供应', 150: '电力工业专用设备', 151: '电力工程施工', 152: '电力生产', 153: '电子和通信测量仪器',
|
|
|
+ 154: '电工、电子专用生产设备', 155: '电影放映', 156: '电气安装', 157: '电气设备', 158: '电气设备修理', 159: '畜牧业服务', 160: '监控设备',
|
|
|
+ 161: '石油制品', 162: '石油和化学工业专用设备', 163: '石油和天然气开采产品', 164: '石油天然气开采专用设备', 165: '研究和试验发展', 166: '社会工作',
|
|
|
+ 167: '社会经济咨询', 168: '科技推广和应用服务业', 169: '科研、医疗、教育用房', 170: '管道和设备安装', 171: '粮油作物和饲料加工设备', 172: '纸、纸制品及印刷品',
|
|
|
+ 173: '纺织原料、毛皮、被服装具', 174: '纺织设备', 175: '绿化管理', 176: '缝纫、服饰、制革和毛皮加工设备', 177: '航空器及其配套设备', 178: '航空客货运输',
|
|
|
+ 179: '航空航天工业专用设备', 180: '节能环保工程施工', 181: '装卸搬运', 182: '计算机和办公设备维修', 183: '计算机设备', 184: '计量标准器具及量具、衡器',
|
|
|
+ 185: '货币处理专用设备', 186: '货币金融服务', 187: '质检技术服务', 188: '资本市场服务', 189: '车辆', 190: '边界勘界和联检专用设备', 191: '运行维护服务',
|
|
|
+ 192: '通信设备', 193: '通用设备修理', 194: '道路货物运输', 195: '邮政专用设备', 196: '邮政业', 197: '采矿业和制造业服务',
|
|
|
+ 198: '铁路、船舶、航空航天等运输设备修理', 199: '铁路、道路、隧道和桥梁工程建筑', 200: '铁路运输设备', 201: '防洪除涝设施管理', 202: '陶瓷制品',
|
|
|
+ 203: '雷达、无线电和卫星导航设备', 204: '非金属矿', 205: '非金属矿物制品工业专用设备', 206: '非金属矿物材料', 207: '食品加工专用设备', 208: '食品及加工盐',
|
|
|
+ 209: '餐饮业', 210: '饮料、酒精及精制茶', 211: '饮料加工设备', 212: '饲养动物及其产品', 213: '黑色金属冶炼及压延产品', 214: '黑色金属矿'}
|
|
|
+ self.industry_dic = {'专业施工': {'大类': '专业施工', '门类': '建筑业'},
|
|
|
+ '专用仪器仪表': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '专用设备修理': {'大类': '金属制品、机械和设备修理业', '门类': '金属制品、机械和设备修理业'},
|
|
|
+ '互联网信息服务': {'大类': '互联网和相关服务', '门类': '信息传输、软件和信息技术服务业'},
|
|
|
+ '互联网安全服务': {'大类': '互联网和相关服务', '门类': '信息传输、软件和信息技术服务业'},
|
|
|
+ '互联网平台': {'大类': '互联网和相关服务', '门类': '信息传输、软件和信息技术服务业'},
|
|
|
+ '互联网接入及相关服务': {'大类': '互联网和相关服务', '门类': '信息传输、软件和信息技术服务业'},
|
|
|
+ '人力资源服务': {'大类': '商务服务业', '门类': '租赁和商务服务业'},
|
|
|
+ '人造原油': {'大类': '炼焦产品、炼油产品', '门类': '零售批发'},
|
|
|
+ '仓储业': {'大类': '装卸搬运和运输代理业', '门类': '交通运输、仓储和邮政业'},
|
|
|
+ '仪器仪表': {'大类': '通用设备', '门类': '零售批发'},
|
|
|
+ '仪器仪表修理': {'大类': '金属制品、机械和设备修理业', '门类': '金属制品、机械和设备修理业'},
|
|
|
+ '会计、审计及税务服务': {'大类': '商务服务业', '门类': '租赁和商务服务业'},
|
|
|
+ '会议、展览及相关服务': {'大类': '商务服务业', '门类': '租赁和商务服务业'},
|
|
|
+ '住宅、商业用房': {'大类': '房屋建筑业', '门类': '建筑业'},
|
|
|
+ '体育场地设施管理': {'大类': '体育', '门类': '文化、体育和娱乐业'},
|
|
|
+ '体育组织': {'大类': '体育', '门类': '文化、体育和娱乐业'},
|
|
|
+ '体育设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '保险服务': {'大类': '保险业', '门类': '金融业'},
|
|
|
+ '信息处理和存储支持服务': {'大类': '软件和信息技术服务业', '门类': '信息传输、软件和信息技术服务业'},
|
|
|
+ '信息技术咨询服务': {'大类': '软件和信息技术服务业', '门类': '信息传输、软件和信息技术服务业'},
|
|
|
+ '信息系统集成和物联网技术服务': {'大类': '软件和信息技术服务业', '门类': '信息传输、软件和信息技术服务业'},
|
|
|
+ '修缮工程': {'大类': '修缮工程', '门类': '建筑业'},
|
|
|
+ '健康咨询': {'大类': '商务服务业', '门类': '租赁和商务服务业'},
|
|
|
+ '公路旅客运输': {'大类': '道路运输业', '门类': '交通运输、仓储和邮政业'},
|
|
|
+ '其他专业咨询与调查': {'大类': '商务服务业', '门类': '租赁和商务服务业'},
|
|
|
+ '其他专业技术服务': {'大类': '专业技术服务业', '门类': '科学研究和技术服务业'},
|
|
|
+ '其他交通运输设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '其他公共设施管理': {'大类': '公共设施管理业', '门类': '水利、环境和公共设施管理业'},
|
|
|
+ '其他土木工程建筑': {'大类': '土木工程建筑业', '门类': '建筑业'},
|
|
|
+ '其他工程服务': {'大类': '工程服务', '门类': '科学研究和技术服务业'},
|
|
|
+ '其他建筑建材': {'大类': '建筑建材', '门类': '零售批发'},
|
|
|
+ '其他运输业': {'大类': '其他运输业', '门类': '交通运输、仓储和邮政业'},
|
|
|
+ '农业和林业机械': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '农业服务': {'大类': '农林牧副渔服务', '门类': '农林牧副渔服务'},
|
|
|
+ '农产品': {'大类': '农林牧渔业产品', '门类': '零售批发'},
|
|
|
+ '农副食品，动、植物油制品': {'大类': '食品、饮料和烟草原料', '门类': '零售批发'},
|
|
|
+ '出版业': {'大类': '新闻和出版业', '门类': '文化、体育和娱乐业'},
|
|
|
+ '办公消耗用品及类似物品': {'大类': '办公消耗用品及类似物品', '门类': '零售批发'},
|
|
|
+ '办公设备': {'大类': '通用设备', '门类': '零售批发'},
|
|
|
+ '化学原料及化学制品': {'大类': '基础化学品及相关产品', '门类': '零售批发'},
|
|
|
+ '化学纤维': {'大类': '基础化学品及相关产品', '门类': '零售批发'},
|
|
|
+ '化学药品和中药专用设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '医疗设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '医药品': {'大类': '医药品', '门类': '零售批发'},
|
|
|
+ '卫星传输服务': {'大类': '电信、广播电视和卫星传输服务', '门类': '信息传输、软件和信息技术服务业'},
|
|
|
+ '卫生': {'大类': '卫生', '门类': '卫生和社会工作'},
|
|
|
+ '印刷服务': {'大类': '商务服务业', '门类': '租赁和商务服务业'},
|
|
|
+ '图书和档案': {'大类': '图书和档案', '门类': '零售批发'},
|
|
|
+ '图书档案设备': {'大类': '通用设备', '门类': '零售批发'},
|
|
|
+ '图书馆与档案馆': {'大类': '文化艺术业', '门类': '文化、体育和娱乐业'},
|
|
|
+ '土地管理业': {'大类': '土地管理业', '门类': '水利、环境和公共设施管理业'},
|
|
|
+ '地质勘查': {'大类': '专业技术服务业', '门类': '科学研究和技术服务业'},
|
|
|
+ '地震服务': {'大类': '专业技术服务业', '门类': '科学研究和技术服务业'},
|
|
|
+ '场馆、站港用房': {'大类': '房屋建筑业', '门类': '建筑业'},
|
|
|
+ '城市公共交通运输': {'大类': '道路运输业', '门类': '交通运输、仓储和邮政业'},
|
|
|
+ '塑料制品、半成品及辅料': {'大类': '橡胶、塑料、玻璃和陶瓷制品', '门类': '零售批发'},
|
|
|
+ '天然石料': {'大类': '建筑建材', '门类': '零售批发'},
|
|
|
+ '娱乐设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '婚姻服务': {'大类': '居民服务业', '门类': '居民服务、修理和其他服务业'},
|
|
|
+ '安全保护服务': {'大类': '商务服务业', '门类': '租赁和商务服务业'},
|
|
|
+ '安全生产设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '家具用具': {'大类': '家具用具', '门类': '零售批发'},
|
|
|
+ '家用电器修理': {'大类': '机动车、电子产品和日用产品修理业', '门类': '居民服务、修理和其他服务业'},
|
|
|
+ '工业、生产用房': {'大类': '房屋建筑业', '门类': '建筑业'},
|
|
|
+ '工业与专业设计及其他专业技术服务': {'大类': '专业技术服务业', '门类': '科学研究和技术服务业'},
|
|
|
+ '工矿工程建筑': {'大类': '土木工程建筑业', '门类': '建筑业'},
|
|
|
+ '工程技术与设计服务': {'大类': '专业技术服务业', '门类': '科学研究和技术服务业'},
|
|
|
+ '工程机械': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '工程监理服务': {'大类': '工程服务', '门类': '科学研究和技术服务业'},
|
|
|
+ '工程评价服务': {'大类': '专业技术服务业', '门类': '科学研究和技术服务业'},
|
|
|
+ '工程造价服务': {'大类': '工程服务', '门类': '科学研究和技术服务业'},
|
|
|
+ '市场调查': {'大类': '商务服务业', '门类': '租赁和商务服务业'},
|
|
|
+ '广告业': {'大类': '商务服务业', '门类': '租赁和商务服务业'},
|
|
|
+ '广播': {'大类': '广播、电视、电影和影视录音制作业', '门类': '文化、体育和娱乐业'},
|
|
|
+ '广播、电视、电影设备': {'大类': '通用设备', '门类': '零售批发'},
|
|
|
+ '广播电视传输服务': {'大类': '电信、广播电视和卫星传输服务', '门类': '信息传输、软件和信息技术服务业'},
|
|
|
+ '废弃资源综合利用业': {'大类': '废弃资源综合利用业', '门类': '废弃资源综合利用业'},
|
|
|
+ '建筑涂料': {'大类': '建筑建材', '门类': '零售批发'},
|
|
|
+ '建筑物、构筑物附属结构': {'大类': '建筑建材', '门类': '零售批发'},
|
|
|
+ '建筑物拆除和场地准备活动': {'大类': '建筑装饰和其他建筑业', '门类': '建筑业'},
|
|
|
+ '建筑装饰和装修业': {'大类': '建筑装饰和其他建筑业', '门类': '建筑业'},
|
|
|
+ '录音制作': {'大类': '广播、电视、电影和影视录音制作业', '门类': '文化、体育和娱乐业'},
|
|
|
+ '影视节目制作': {'大类': '广播、电视、电影和影视录音制作业', '门类': '文化、体育和娱乐业'},
|
|
|
+ '房地产中介服务': {'大类': '房地产业', '门类': '房地产业'},
|
|
|
+ '房地产开发经营': {'大类': '房地产业', '门类': '房地产业'},
|
|
|
+ '房地产租赁经营': {'大类': '房地产业', '门类': '房地产业'},
|
|
|
+ '房屋租赁': {'大类': '租赁业', '门类': '租赁和商务服务业'},
|
|
|
+ '招标代理': {'大类': '专业技术服务业', '门类': '科学研究和技术服务业'},
|
|
|
+ '探矿、采矿、选矿和造块设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '政法、检测专用设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '教育服务': {'大类': '教育服务', '门类': '教育'},
|
|
|
+ '教育设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '文体设备和用品出租': {'大类': '租赁业', '门类': '租赁和商务服务业'},
|
|
|
+ '文物及非物质文化遗产保护': {'大类': '文化艺术业', '门类': '文化、体育和娱乐业'},
|
|
|
+ '文物和陈列品': {'大类': '文物和陈列品', '门类': '零售批发'},
|
|
|
+ '文艺创作与表演': {'大类': '文化艺术业', '门类': '文化、体育和娱乐业'},
|
|
|
+ '文艺设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '新闻业': {'大类': '新闻和出版业', '门类': '文化、体育和娱乐业'},
|
|
|
+ '旅行社及相关服务': {'大类': '商务服务业', '门类': '租赁和商务服务业'},
|
|
|
+ '日杂用品': {'大类': '日杂用品', '门类': '零售批发'},
|
|
|
+ '有色金属冶炼及压延产品': {'大类': '建筑建材', '门类': '零售批发'},
|
|
|
+ '有色金属矿': {'大类': '矿与矿物', '门类': '零售批发'},
|
|
|
+ '木材、板材等': {'大类': '建筑建材', '门类': '零售批发'},
|
|
|
+ '木材采集和加工设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '机械设备': {'大类': '通用设备', '门类': '零售批发'},
|
|
|
+ '机械设备经营租赁': {'大类': '租赁业', '门类': '租赁和商务服务业'},
|
|
|
+ '林业产品': {'大类': '农林牧渔业产品', '门类': '零售批发'},
|
|
|
+ '林业服务': {'大类': '农林牧副渔服务', '门类': '农林牧副渔服务'},
|
|
|
+ '架线和管道工程建筑': {'大类': '土木工程建筑业', '门类': '建筑业'},
|
|
|
+ '核工业专用设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '橡胶制品': {'大类': '橡胶、塑料、玻璃和陶瓷制品', '门类': '零售批发'},
|
|
|
+ '殡葬服务': {'大类': '居民服务业', '门类': '居民服务、修理和其他服务业'},
|
|
|
+ '殡葬设备及用品': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '气象服务': {'大类': '专业技术服务业', '门类': '科学研究和技术服务业'},
|
|
|
+ '水上交通运输设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '水上运输业': {'大类': '水上运输业', '门类': '交通运输、仓储和邮政业'},
|
|
|
+ '水利和水运工程建筑': {'大类': '土木工程建筑业', '门类': '建筑业'},
|
|
|
+ '水工机械': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '水文服务': {'大类': '水利管理业', '门类': '水利、环境和公共设施管理业'},
|
|
|
+ '水资源管理': {'大类': '水利管理业', '门类': '水利、环境和公共设施管理业'},
|
|
|
+ '污水处理及其再生利用': {'大类': '水的生产和供应业', '门类': '电力、热力、燃气及水生产和供应业'},
|
|
|
+ '汽车、摩托车修理与维护': {'大类': '机动车、电子产品和日用产品修理业', '门类': '居民服务、修理和其他服务业'},
|
|
|
+ '法律服务': {'大类': '商务服务业', '门类': '租赁和商务服务业'},
|
|
|
+ '洗染服务': {'大类': '居民服务业', '门类': '居民服务、修理和其他服务业'},
|
|
|
+ '测绘地理信息服务': {'大类': '专业技术服务业', '门类': '科学研究和技术服务业'},
|
|
|
+ '海洋仪器设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '海洋工程建筑': {'大类': '土木工程建筑业', '门类': '建筑业'},
|
|
|
+ '海洋服务': {'大类': '专业技术服务业', '门类': '科学研究和技术服务业'},
|
|
|
+ '消防设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '清洁服务': {'大类': '其他服务业', '门类': '居民服务、修理和其他服务业'},
|
|
|
+ '渔业产品': {'大类': '农林牧渔业产品', '门类': '零售批发'},
|
|
|
+ '渔业服务': {'大类': '农林牧副渔服务', '门类': '农林牧副渔服务'},
|
|
|
+ '炼焦和金属冶炼轧制设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '烟草加工设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '热力生产和供应': {'大类': '电力、热力生产和供应业', '门类': '电力、热力、燃气及水生产和供应业'},
|
|
|
+ '焦炭及其副产品': {'大类': '炼焦产品、炼油产品', '门类': '零售批发'},
|
|
|
+ '煤炭采选产品': {'大类': '矿与矿物', '门类': '零售批发'},
|
|
|
+ '燃气生产和供应业': {'大类': '燃气生产和供应业', '门类': '电力、热力、燃气及水生产和供应业'},
|
|
|
+ '物业管理': {'大类': '房地产业', '门类': '房地产业'},
|
|
|
+ '特种用途动、植物': {'大类': '农林牧渔业产品', '门类': '零售批发'},
|
|
|
+ '环保咨询': {'大类': '商务服务业', '门类': '租赁和商务服务业'},
|
|
|
+ '环境与生态监测检测服务': {'大类': '专业技术服务业', '门类': '科学研究和技术服务业'},
|
|
|
+ '环境污染防治设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '环境治理业': {'大类': '生态保护和环境治理业', '门类': '水利、环境和公共设施管理业'},
|
|
|
+ '玻璃及其制品': {'大类': '橡胶、塑料、玻璃和陶瓷制品', '门类': '零售批发'},
|
|
|
+ '理发及美容服务': {'大类': '居民服务业', '门类': '居民服务、修理和其他服务业'},
|
|
|
+ '生态保护': {'大类': '生态保护和环境治理业', '门类': '水利、环境和公共设施管理业'},
|
|
|
+ '电信': {'大类': '电信、广播电视和卫星传输服务', '门类': '信息传输、软件和信息技术服务业'},
|
|
|
+ '电力、城市燃气、蒸汽和热水、水': {'大类': '电力、城市燃气、蒸汽和热水、水', '门类': '零售批发'},
|
|
|
+ '电力供应': {'大类': '电力、热力生产和供应业', '门类': '电力、热力、燃气及水生产和供应业'},
|
|
|
+ '电力工业专用设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '电力工程施工': {'大类': '土木工程建筑业', '门类': '建筑业'},
|
|
|
+ '电力生产': {'大类': '电力、热力生产和供应业', '门类': '电力、热力、燃气及水生产和供应业'},
|
|
|
+ '电子和通信测量仪器': {'大类': '通用设备', '门类': '零售批发'},
|
|
|
+ '电工、电子专用生产设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '电影放映': {'大类': '广播、电视、电影和影视录音制作业', '门类': '文化、体育和娱乐业'},
|
|
|
+ '电气安装': {'大类': '建筑安装业', '门类': '建筑业'},
|
|
|
+ '电气设备': {'大类': '通用设备', '门类': '零售批发'},
|
|
|
+ '电气设备修理': {'大类': '金属制品、机械和设备修理业', '门类': '金属制品、机械和设备修理业'},
|
|
|
+ '畜牧业服务': {'大类': '农林牧副渔服务', '门类': '农林牧副渔服务'},
|
|
|
+ '监控设备': {'大类': '通用设备', '门类': '零售批发'},
|
|
|
+ '石油制品': {'大类': '炼焦产品、炼油产品', '门类': '零售批发'},
|
|
|
+ '石油和化学工业专用设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '石油和天然气开采产品': {'大类': '矿与矿物', '门类': '零售批发'},
|
|
|
+ '石油天然气开采专用设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '研究和试验发展': {'大类': '研究和试验发展', '门类': '科学研究和技术服务业'},
|
|
|
+ '社会工作': {'大类': '社会工作', '门类': '卫生和社会工作'},
|
|
|
+ '社会经济咨询': {'大类': '商务服务业', '门类': '租赁和商务服务业'},
|
|
|
+ '科技推广和应用服务业': {'大类': '科技推广和应用服务业', '门类': '科学研究和技术服务业'},
|
|
|
+ '科研、医疗、教育用房': {'大类': '房屋建筑业', '门类': '建筑业'},
|
|
|
+ '管道和设备安装': {'大类': '建筑安装业', '门类': '建筑业'},
|
|
|
+ '粮油作物和饲料加工设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '纸、纸制品及印刷品': {'大类': '纸、纸制品及印刷品', '门类': '零售批发'},
|
|
|
+ '纺织原料、毛皮、被服装具': {'大类': '纺织原料、毛皮、被服装具', '门类': '零售批发'},
|
|
|
+ '纺织设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '绿化管理': {'大类': '公共设施管理业', '门类': '水利、环境和公共设施管理业'},
|
|
|
+ '缝纫、服饰、制革和毛皮加工设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '航空器及其配套设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '航空客货运输': {'大类': '航空运输业', '门类': '交通运输、仓储和邮政业'},
|
|
|
+ '航空航天工业专用设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '节能环保工程施工': {'大类': '土木工程建筑业', '门类': '建筑业'},
|
|
|
+ '装卸搬运': {'大类': '装卸搬运和运输代理业', '门类': '交通运输、仓储和邮政业'},
|
|
|
+ '计算机和办公设备维修': {'大类': '机动车、电子产品和日用产品修理业', '门类': '居民服务、修理和其他服务业'},
|
|
|
+ '计算机设备': {'大类': '通用设备', '门类': '零售批发'},
|
|
|
+ '计量标准器具及量具、衡器': {'大类': '通用设备', '门类': '零售批发'},
|
|
|
+ '货币处理专用设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '货币金融服务': {'大类': '货币金融服务', '门类': '金融业'},
|
|
|
+ '质检技术服务': {'大类': '专业技术服务业', '门类': '科学研究和技术服务业'},
|
|
|
+ '资本市场服务': {'大类': '资本市场服务', '门类': '金融业'},
|
|
|
+ '车辆': {'大类': '通用设备', '门类': '零售批发'},
|
|
|
+ '边界勘界和联检专用设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '运行维护服务': {'大类': '软件和信息技术服务业', '门类': '信息传输、软件和信息技术服务业'},
|
|
|
+ '通信设备': {'大类': '通用设备', '门类': '零售批发'},
|
|
|
+ '通用设备修理': {'大类': '金属制品、机械和设备修理业', '门类': '金属制品、机械和设备修理业'},
|
|
|
+ '道路货物运输': {'大类': '道路运输业', '门类': '交通运输、仓储和邮政业'},
|
|
|
+ '邮政专用设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '邮政业': {'大类': '邮政业', '门类': '交通运输、仓储和邮政业'},
|
|
|
+ '采矿业和制造业服务': {'大类': '采矿业和制造业服务', '门类': '农林牧副渔服务'},
|
|
|
+ '铁路、船舶、航空航天等运输设备修理': {'大类': '金属制品、机械和设备修理业', '门类': '金属制品、机械和设备修理业'},
|
|
|
+ '铁路、道路、隧道和桥梁工程建筑': {'大类': '土木工程建筑业', '门类': '建筑业'},
|
|
|
+ '铁路运输设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '防洪除涝设施管理': {'大类': '水利管理业', '门类': '水利、环境和公共设施管理业'},
|
|
|
+ '陶瓷制品': {'大类': '橡胶、塑料、玻璃和陶瓷制品', '门类': '零售批发'},
|
|
|
+ '雷达、无线电和卫星导航设备': {'大类': '通用设备', '门类': '零售批发'},
|
|
|
+ '非金属矿': {'大类': '矿与矿物', '门类': '零售批发'},
|
|
|
+ '非金属矿物制品工业专用设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '非金属矿物材料': {'大类': '建筑建材', '门类': '零售批发'},
|
|
|
+ '食品加工专用设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '食品及加工盐': {'大类': '食品、饮料和烟草原料', '门类': '零售批发'},
|
|
|
+ '餐饮业': {'大类': '餐饮业', '门类': '住宿和餐饮业'},
|
|
|
+ '饮料、酒精及精制茶': {'大类': '食品、饮料和烟草原料', '门类': '零售批发'},
|
|
|
+ '饮料加工设备': {'大类': '专用设备', '门类': '零售批发'},
|
|
|
+ '饲养动物及其产品': {'大类': '农林牧渔业产品', '门类': '零售批发'},
|
|
|
+ '黑色金属冶炼及压延产品': {'大类': '建筑建材', '门类': '零售批发'},
|
|
|
+ '黑色金属矿': {'大类': '矿与矿物', '门类': '零售批发'}}
|
|
|
+ self.sess = tf.Session(graph=tf.Graph())
|
|
|
+ self.get_model()
|
|
|
+
|
|
|
+ with open(os.path.dirname(__file__)+'/industry_rule_kw_json/tw_industry_keyword_org/tw_industry_keyword_org.json', 'r',
|
|
|
+ encoding='utf-8') as fp1:
|
|
|
+ self.json_data_industry = json.load(fp1)
|
|
|
+ with open(os.path.dirname(__file__)+'/industry_rule_kw_json/tw_company_classification_keyword/tw_company_classification_keyword.json', 'r',
|
|
|
+ encoding='utf-8') as fp2:
|
|
|
+ self.json_data_company = json.load(fp2)
|
|
|
+ with open(os.path.dirname(__file__)+'/industry_rule_kw_json/tw_custom_keyword/tw_custom_keyword.json', 'r', encoding='utf-8') as fp3:
|
|
|
+ self.json_data_custom = json.load(fp3)
|
|
|
+
|
|
|
+
|
|
|
+ def get_model(self):
|
|
|
+ with self.sess.as_default() as sess:
|
|
|
+ with self.sess.graph.as_default():
|
|
|
+ meta_graph_def = tf.saved_model.loader.load(sess,
|
|
|
+ tags=['serve'],
|
|
|
+ export_dir=os.path.dirname(__file__)+'/industry_model')
|
|
|
+ signature_key = tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY
|
|
|
+ signature_def = meta_graph_def.signature_def
|
|
|
+
|
|
|
+ self.title = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs['title'].name)
|
|
|
+ self.project = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs['project'].name)
|
|
|
+ self.product = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs['product'].name)
|
|
|
+ self.outputs = sess.graph.get_tensor_by_name(signature_def[signature_key].outputs['outputs'].name)
|
|
|
+
|
|
|
+ def text2array(self, text, tenderee='', maxSententLen=20):
|
|
|
+ tenderee = tenderee.replace('(', '（').replace(')', '）')
|
|
|
+ text = text.replace('(', '（').replace(')', '）')
|
|
|
+ text = re.sub(
|
|
|
+ '(废标|终止|综?合?评审|评标|开标|资审|履约|验收|成交|中标人?|中选人?|单一来源|合同|候选人|结果|变更|更正|答疑|澄清|意向|需求|采购|招标|询比?价|磋商|谈判|比选|比价|竞价|议价)的?(公告|预告|公示)?|关于为?|选取|定点|直接|邀请函?|通知书?|备案|公开|公示|公告|记录|竞争性',
|
|
|
+ '', text)
|
|
|
+ text = text.replace(tenderee, '')
|
|
|
+ text = ' ' if text=="" else text
|
|
|
+ words_docs_list = fool.cut(text)
|
|
|
+ words_docs_list = [[it for it in l if re.search('^[\u4e00-\u9fa5]+$', it)][-maxSententLen:] for l in words_docs_list]
|
|
|
+ array = embedding(words_docs_list, shape=(len(words_docs_list), maxSententLen, 128))
|
|
|
+ return array
|
|
|
+
|
|
|
+ def process(self, title, project, product, tenderee):
|
|
|
+ return self.text2array(title, tenderee), self.text2array(project, tenderee), self.text2array(product)
|
|
|
+
|
|
|
+ def predict_model(self, title, project, product, tenderee=''):
|
|
|
+ title_array, project_array, product_array = self.process(title, project, product, tenderee)
|
|
|
+ rs = self.sess.run(self.outputs,
|
|
|
+ feed_dict={
|
|
|
+ self.title:title_array,
|
|
|
+ self.project:project_array,
|
|
|
+ self.product:product_array
|
|
|
+ }
|
|
|
+ )
|
|
|
+ pred = np.argmax(rs[0])
|
|
|
+ return self.id2lb[pred], rs[0][pred]
|
|
|
+ # # 返回top2 结果
|
|
|
+ # pred_list = np.argsort(-rs[0])
|
|
|
+ # return self.id2lb[pred_list[0]], self.id2lb[pred_list[1]], rs[0][pred_list[0]], rs[0][pred_list[1]]
|
|
|
+
|
|
|
+ def predict_rule(self, doctitle, tenderee, win_tenderer, project_name, product):
|
|
|
+ doctitle = doctitle if doctitle else ''
|
|
|
+ tenderee = tenderee if tenderee else ''
|
|
|
+ win_tenderer = win_tenderer if win_tenderer else ''
|
|
|
+ project_name = project_name if project_name else ''
|
|
|
+ product = product if product else ''
|
|
|
+
|
|
|
+ text_ind = (doctitle + project_name + product).replace(tenderee, '')
|
|
|
+ text_com = win_tenderer
|
|
|
+
|
|
|
+ length_ind_text = len(text_ind) + 1
|
|
|
+ length_com_text = len(text_com) + 1
|
|
|
+ # print(text)
|
|
|
+
|
|
|
+ dic_res = {} # 行业分类字典
|
|
|
+ score_lst = [] # 得分列表
|
|
|
+ word_lst = [] # 关键词列表
|
|
|
+
|
|
|
+ # 主要内容关键词
|
|
|
+ if text_ind:
|
|
|
+ # logging.info("data_ind%s"%str(_json_data_industry[0]))
|
|
|
+ for data_industry in self.json_data_industry:
|
|
|
+ industry = data_industry['xiaolei']
|
|
|
+ key_word = data_industry['key_word']
|
|
|
+ key_word_2 = data_industry['key_word2']
|
|
|
+ power = float(data_industry['power']) if data_industry['power'] else 0
|
|
|
+ this_score = power * (text_ind.count(key_word) * len(key_word) / length_ind_text)
|
|
|
+
|
|
|
+ if key_word_2:
|
|
|
+ # key_word_compose = key_word + "+" + key_word_2
|
|
|
+ if text_ind.count(key_word_2) == 0:
|
|
|
+ this_score = 0
|
|
|
+
|
|
|
+ if this_score > 0:
|
|
|
+ # print(industry,key_word,this_score)
|
|
|
+ if industry in dic_res.keys():
|
|
|
+ dic_res[industry] += this_score
|
|
|
+ else:
|
|
|
+ dic_res[industry] = this_score
|
|
|
+
|
|
|
+ if key_word not in word_lst:
|
|
|
+ word_lst.append(key_word)
|
|
|
+
|
|
|
+ # 供应商关键词
|
|
|
+ if text_com:
|
|
|
+
|
|
|
+ for data_company in self.json_data_company:
|
|
|
+ industry = data_company['industry_type']
|
|
|
+ key_word = data_company['company_word']
|
|
|
+ power = float(data_company['industry_rate']) if data_company['industry_rate'] else 0
|
|
|
+ this_score = power * (text_com.count(key_word) * len(key_word) / length_com_text)
|
|
|
+
|
|
|
+ if this_score > 0:
|
|
|
+ # print(industry,key_word,this_score)
|
|
|
+ if industry in dic_res.keys():
|
|
|
+ dic_res[industry] += this_score
|
|
|
+ else:
|
|
|
+ dic_res[industry] = this_score
|
|
|
+
|
|
|
+ if key_word not in word_lst:
|
|
|
+ word_lst.append(key_word)
|
|
|
+
|
|
|
+ # 自定义关键词
|
|
|
+ if text_ind:
|
|
|
+
|
|
|
+ custom_ind = [
|
|
|
+ ['tenderee', '医院|疾病预防', ['设备', '系统', '器'], '医疗设备'],
|
|
|
+ ['tenderee', '学校|大学|小学|中学|学院|幼儿园', ['设备', '器'], '教育设备'],
|
|
|
+ ['tenderee', '学校|大学|小学|中学|学院|幼儿园|医院', ['工程'], '科研、医疗、教育用房'],
|
|
|
+ ['tenderee', '供电局|电网|国网|电力|电厂|粤电', ['设备', '器', '物资'], '电力工业专用设备'],
|
|
|
+ ['tenderee', '公安|法院|检察院', ['设备', '器'], '政法、检测专用设备'],
|
|
|
+ ['tenderee', '^中铁|^中交|^中建|中国建筑', ['材料'], '其他建筑建材'],
|
|
|
+ ['doctextcon', '信息技术服务|系统开发|信息化|信息系统', ['监理'], '信息技术咨询服务'],
|
|
|
+ ['doctextcon', '工程', ['消防'], '专业施工'],
|
|
|
+ ['doctextcon', '铁路|航空|船舶|航天|广铁', ['维修'], '铁路、船舶、航空航天等运输设备修理'],
|
|
|
+ ['doctextcon', '设备|仪|器', ['租赁'], '机械设备经营租赁'],
|
|
|
+ ['doctextcon', '交通|铁路|公路|道路|桥梁', ['工程'], '铁路、道路、隧道和桥梁工程建筑'],
|
|
|
+ ['win_tenderer', '电力', ['设备', '器'], '电力工业专用设备'],
|
|
|
+ ['win_tenderer', '信息|网络科技', ['系统'], '信息系统集成和物联网技术服务'],
|
|
|
+ ['tenderee,doctextcon', '铁路|广铁|铁道', ['设备', '器', '物资', '材料', '铁路'], '铁路运输设备'],
|
|
|
+ ]
|
|
|
+
|
|
|
+ for data_custom in self.json_data_custom:
|
|
|
+ industry_custom = data_custom['industry']
|
|
|
+ key_word = data_custom['company_word']
|
|
|
+ power = float(data_custom['industry_rate'])
|
|
|
+ for k in range(len(custom_ind)):
|
|
|
+ subject = ''
|
|
|
+ if 'tenderee' in custom_ind[k][0]:
|
|
|
+ subject += tenderee
|
|
|
+ if 'win_tenderer' in custom_ind[k][0]:
|
|
|
+ subject += win_tenderer
|
|
|
+ if 'doctextcon' in custom_ind[k][0]:
|
|
|
+ subject += text_ind
|
|
|
+
|
|
|
+ ptn = custom_ind[k][1]
|
|
|
+ # print('ptn',ptn)
|
|
|
+ if re.search(ptn, subject) and industry_custom in custom_ind[k][2]:
|
|
|
+ industry = custom_ind[k][3]
|
|
|
+ else:
|
|
|
+ continue
|
|
|
+
|
|
|
+ this_score = power * (text_ind.count(key_word) * len(key_word) / len(subject))
|
|
|
+
|
|
|
+ if this_score > 0:
|
|
|
+ # print(industry,key_word,this_score)
|
|
|
+ if industry in dic_res.keys():
|
|
|
+ dic_res[industry] += this_score
|
|
|
+ else:
|
|
|
+ dic_res[industry] = this_score
|
|
|
+
|
|
|
+ if key_word not in word_lst:
|
|
|
+ word_lst.append(key_word)
|
|
|
+ sort_res = sorted(dic_res.items(), key=lambda x: x[1], reverse=True)
|
|
|
+ lst_res = [s[0] for s in sort_res]
|
|
|
+ score_lst = [str(round(float(s[1]), 2)) for s in sort_res]
|
|
|
+ if len(lst_res) > 0:
|
|
|
+ return lst_res, score_lst, word_lst
|
|
|
+ else:
|
|
|
+ return [""], [], []
|
|
|
+
|
|
|
+ def predict_merge(self, pinmu_type, industry_lst):
|
|
|
+ '''
|
|
|
+ 通过一系列规则最终决定使用模型还是规则的结果
|
|
|
+ :param pinmu_type: 模型预测类别
|
|
|
+ :param industry_lst: 规则预测类别列表
|
|
|
+ :return:
|
|
|
+ '''
|
|
|
+ industry_type = industry_lst[0]
|
|
|
+ if industry_type == "":
|
|
|
+ return pinmu_type
|
|
|
+ if industry_type == '专用设备修理' and re.search('修理|维修|装修|修缮', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '其他土木工程建筑' and re.search('工程|建筑|用房|施工|安装|质检|其他专业咨询与调查', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif pinmu_type == '专用设备修理' and re.search('工程|修理', industry_type):
|
|
|
+ final_type = industry_type
|
|
|
+ elif pinmu_type == '信息系统集成和物联网技术服务' and re.search('卫星传输|信息处理和存储支持服务|信息技术咨询服务|运行维护服务|其他专业技术服务|医疗设备|医药品',
|
|
|
+ industry_type):
|
|
|
+ final_type = industry_type
|
|
|
+ elif industry_type == '仪器仪表' and re.search('仪器|器具|医疗设备', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '医药品' and re.search('医疗设备', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '医药品' and re.search('医疗设备', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif re.search('设备', industry_type) and re.search('修理|维修', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '社会工作' and re.search('工程', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '信息系统集成和物联网技术服务' and re.search('信息处理|设备', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '研究和试验发展' and re.search('其他专业咨询与调查|质检技术服务|信息系统集成|其他工程服务', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '其他专业咨询与调查' and re.search('工程造价服务', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '广告业' and re.search('印刷服务|影视节目制作|信息系统', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '清洁服务' and re.search('工程|环境污染防治设备|修理', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '其他公共设施管理' and re.search('信息系统', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '其他专业技术服务' and re.search('工程技术与设计服务|质检技术服务|环境与生态监测检测服务', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '机械设备经营租赁' and re.search('电信', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '货币金融服务' and re.search('信息系统集成和物联网技术服务', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '体育场地设施管理' and re.search('体育设备', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '安全保护服务' and re.search('信息系统|监控设备|互联网安全服务', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '互联网接入及相关服务' and re.search('通信设备', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type == '卫生' and re.search('医疗设备|信息系统', pinmu_type):
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif pinmu_type == '研究和试验发展' and re.search('其他工程服务', industry_type):
|
|
|
+ final_type = industry_type
|
|
|
+ elif pinmu_type == '办公设备' and re.search('教育设备', industry_type):
|
|
|
+ final_type = industry_type
|
|
|
+ elif re.search('车辆|机械设备经营租赁', pinmu_type) and re.search('公路旅客运输', industry_type):
|
|
|
+ final_type = industry_type
|
|
|
+ elif len(industry_lst) > 1 and pinmu_type == industry_lst[1] and re.search('会计|法律|物业|家具|印刷|互联网安全',
|
|
|
+ industry_type) == None \
|
|
|
+ and re.search('其他|人力资源服务', pinmu_type) == None:
|
|
|
+ final_type = pinmu_type
|
|
|
+ elif industry_type != "":
|
|
|
+ final_type = industry_type
|
|
|
+ else:
|
|
|
+ final_type = pinmu_type
|
|
|
+ return final_type
|
|
|
+
|
|
|
+ def predict(self, title, project, product, prem):
|
|
|
+ def get_ree_win(prem):
|
|
|
+ tenderee = ""
|
|
|
+ win_tenderer = ""
|
|
|
+ try:
|
|
|
+ for v in prem[0]['prem'].values():
|
|
|
+ for link in v['roleList']:
|
|
|
+ if link['role_name'] == 'tenderee' and tenderee == "":
|
|
|
+ tenderee = link['role_text']
|
|
|
+ elif link['role_name'] == 'win_tenderer' and win_tenderer == "":
|
|
|
+ win_tenderer = link['role_text']
|
|
|
+ except Exception as e:
|
|
|
+ print('解析prem 获取招标人、中标人出错')
|
|
|
+ return tenderee, win_tenderer
|
|
|
+ tenderee, win_tenderer = get_ree_win(prem)
|
|
|
+ result_model, prob = self.predict_model(title, project, product, tenderee)
|
|
|
+ industry_lst, score_lst, word_lst = self.predict_rule(title, tenderee, win_tenderer, project, product)
|
|
|
+ final_type = self.predict_merge(result_model, industry_lst)
|
|
|
+ # print('模型：%s；规则：%s；最终：%s'%(result_model, industry_lst[0], final_type))
|
|
|
+ # return {'industry': final_type}
|
|
|
+ return {'industry': {
|
|
|
+ 'class_name': final_type,
|
|
|
+ 'subclass': self.industry_dic[final_type]['大类'],
|
|
|
+ 'class': self.industry_dic[final_type]['门类']
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
|
|
|
def getSavedModel():
|
|
|
#predictor = FormPredictor()
|
