【在线部署一个机器学习模型】

from flask import Flask, jsonify, request
import joblib

# 创建Flask应用程序
app = Flask(__name__)

# 加载训练好的模型
model = joblib.load('model.pkl')

# 定义预测路由
@app.route('/predict', methods=['POST'])
def predict():
    # 获取POST请求中的数据
    data = request.get_json(force=True)
    
    # 对请求的数据进行预测
    prediction = model.predict([data['input']])
    
    # 返回预测结果
    return jsonify({'prediction': prediction.tolist()})

# 运行应用程序
if __name__ == '__main__':
    app.run(debug=True)

python app.py