(四)机器学习模型的api调用之特征查询

根据模型查询特征接口:
http://ip:port//model/features
post
{
“modelFilePath”:"/data/model/lr.pkl"
}

后台代码实现:

from flask import request
import joblib
import time

modelDir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))+ '/model/'

#模型文件路径校验,参数不存在使用默认模型
def modelFilePathCheck(request):
    if  keyIsExist(request,'modelFilePath'):
        modelFilePath = request['modelFilePath']
    else:
        modelFilePath = modelDir + 'default/lr.pkl'
    return modelFilePath

#模型特征查询接口    
@app.route('/model/features',methods=['POST'])
def features_data():
    start = time.time()
    try:
        logger.info(str(request.get_data()))
        request_data = request.get_json() #获取传入数据
        modelFilePath = modelFilePathCheck(request_data)
        if os.path.isfile(modelFilePath):
            #调用model文件
            model = joblib.load(modelFilePath)
            bins = model.bins
            result = {}
            result['code'] = '00000'
            result['features'] = list(bins.keys())
            end = time.time()
            logging.info('特征查询耗时:%s',end-start) #计算下接口耗时
            return jsonify(result)

        return jsonify(code10009)

    except KeyError as e:
        logger.info(e)
        code10001['errorMsg']='输入参数错误:' + str(e)
        return jsonify(code10001)


(一)机器学习模型的api调用之初探
(二)机器学习模型的api调用之日志打印
(三)机器学习模型的api调用之异常控制
(四)机器学习模型的api调用之特征查询
(五)机器学习模型的api调用之模型执行

源代码地址

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值