flask 为prophet 模型提供API接口

from flask import Flask, jsonify, request
from flask_cors import CORS, cross_origin
app = Flask(__name__)
CORS(app)

import pandas as pd
import numpy as np
from fbprophet import Prophet

df = pd.read_csv('./data/476.csv')
df =df[1:]

m = Prophet()
m=m.fit(df)

import pickle
with open('399model.pckl', 'wb') as fout:
    pickle.dump(m, fout)

with open('399model.pckl', 'rb') as fin:
    m2 = pickle.load(fin)

@app.route("/prophetapi",methods=['get'])
def predict():
    #horizon = int(request.json['horizon'])
    horizon = int(request.args.get('horizon'))
    
    future2 = m2.make_future_dataframe(periods=horizon,freq='BH')
    forecast2 = m2.predict(future2)
    
    data = forecast2[['ds', 'yhat']][-horizon:]
    
    ret = data.to_json(orient='records', date_format='iso')
    
    return ret
# running REST interface, port=3000 for direct test
if __name__ == "__main__":
    app.run(debug=False, host='127.0.0.1', port=3000)