# -*- coding: utf-8 -*- """``IndustryPredictor`` — 行业分类模型+规则混合预测。 Phase 5 从 ``interface/predictor.py``（约 6235-6769 行）迁出。 原 ``from common.Utils import *`` / ``from common.nerUtils import *`` 已替换为显式 import；``os.path.dirname(__file__)`` 路径引用替换为 ``predictors._common.INTERFACE_DIR``。 """ from __future__ import absolute_import import os import re import json import numpy as np import tensorflow as tf from BiddingKG.dl.foolnltk import selffool from BiddingKG.dl.model_runtime.embed import embedding from BiddingKG.dl.predictors._common import INTERFACE_DIR __all__ = ["IndustryPredictor"] class IndustryPredictor(): def __init__(self,): self.model_path = os.path.join(INTERFACE_DIR, '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.join(INTERFACE_DIR, '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.join(INTERFACE_DIR, '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.join(INTERFACE_DIR, 'industry_rule_kw_json', 'tw_custom_keyword', 'tw_custom_keyword.json'), 'r', encoding='utf-8') as fp3: self.json_data_custom = json.load(fp3) '''下面补充行业关键词''' d = {'id': 5592, 'pingmu': '工程', 'menlei': '建筑业', 'dalei': '建筑装饰和其他建筑业', 'xiaolei': '建筑物拆除和场地准备活动', 'key_word': '围蔽', 'key_word2': None, 'power': '6.00'} self.json_data_industry.append(d) 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.join(INTERFACE_DIR, '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 = selffool.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_ind = text_ind.replace('墙面粉刷', '墙面 粉刷') 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, product_attrs): 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 if product_attrs[0]['product_attrs'].get('data', [])!=[]: # 20250116 如果产品属性提取到产品，替换产品 products = [d['product'] for d in product_attrs[0]['product_attrs']['data']] product = ' '.join(products) 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]['门类'] } }