本文讲一下利用tornado框架简单部署lightgbm模型服务的demo,具体步骤如下:
首先,准备好线下训练好的lightgbm模型文件,本文使用iris数据集训练得到。
然后,部署代码deploy.py如下:
import tornado.ioloop
import tornado.web
import json
import joblib
import numpy as np
class MainHandler(tornado.web.RequestHandler):
def get(self):
self.write("Hello, 666666")
def post(self):
data = self.request.body.decode("utf-8")
data = json.loads(data)
# # ==== 单条数据测试
# data = np.array(data["data"]).reshape(1, 4) # 单条数据测试
# predict_lbl = clf.predict(data)[0]
#
# msg = {"label": int(predict_lbl), "code": 200}
# self.write(json.dumps(msg))
# # ====================
# ==== 多条数据测试
data = np.array(data