Rückgabe csv als JSON in Flask

Question

Ich möchte in JSON die neuen Datenprognosen (new_data.csv) eines maschinellen Lernmodells, das in joblib gespeichert wurde, zurückgeben.

# -*- 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

können Sie verwenden 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)