ContentExtract/module/settings.py

# -*- coding: utf-8 -*-
"""
Created on Fri Nov 23 10:56:42 2018

@author: LIN Jinxing
"""
import logging
# 初始化redis数据库连接参数
REDIS_HOST = "192.168.2.101"
REDIS_PORT = 6379
REDIS_DB = 9
REDIS_PASS = "bidikg_$*2018"

# 初始化Redis中队列名称和程序参数
CONTENT_QUEUE = "ContentExtract_content"
ERROR_QUEUE = "ContentExtract_error"
BATCH_SIZE = 2
SERVER_SLEEP = 0.002
CLIENT_SLEEP = 0.002

# 初始化 MODEL SERVER 日志配置
LOG_FORMAT = "%(asctime)s - %(levelname)s - %(message)s"
DATE_FORMAT = "%m/%d/%Y %H:%M:%S %p"
LOG_LEVEL = logging.DEBUG
LOG_FILE = 'logs/model_server.log'

#模型的输入shape
MODEL_ROLE_INPUT_SHAPE = (2,10,128)
#MODEL_MONEY_INPUT_SHAPE = (2,10,128)
MODEL_MONEY_INPUT_SHAPE = (3,100,60)
MODEL_PERSON_INPUT_SHAPE = (2,10,128)