Flask에서 json으로 csv를 반환하십시오.

Question

joblib에 저장된 기계 학습 모델의 새로운 데이터 예측 (new_data.csv)을 JSON으로 반환하고 싶습니다.

# -*- coding: utf-8 -*- 
from flask import Flask,request, jsonify 
from sklearn.externals import joblib 
import pandas as pd 

app = Flask(__name__) 

@app.route('/predict', methods=['GET', 'POST']) 

def predict() :  
    json_ = request.json 
    new = pd.read_csv('new_data.csv') 
    json_vector = new.transform(json_) 
    query = pd.DataFrame(json_vector) 
    prediction = regr.predict(query) 
    return json.dumps({'prediction': list({{prediction}})}) 

if __name__ == '__main__' : 
    regr = joblib.load('model.pkl') 
    app.run(port=8080, debug=True) 

Answer 1

당신은 jsonify 사용할 수 있습니다

from flask import jsonify 

def predict() :  
    json_ = request.json 
    new = pd.read_csv('new_data.csv') 
    json_vector = new.transform(json_) 
    query = pd.DataFrame(json_vector) 
    prediction = regr.predict(query) 
    data = {'prediction': list({{prediction}})} 
    return jsonify(data)