forked from a-djebali/flask-machine-learning-resful
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathapp.py
More file actions
26 lines (19 loc) · 703 Bytes
/
app.py
File metadata and controls
26 lines (19 loc) · 703 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import numpy as np
from flask import Flask, abort, jsonify, request
import cPickle as pickle
random_forest_model = pickle.load(open("rfc.pkl","rb"))
app = Flask(__name__)
@app.route('/predict_api', methods=['POST'])
def predict():
# Error checking
data = request.get_json(force=True)
# Convert JSON to numpy array
predict_request = [data['sl'],data['sw'],data['pl'],data['pw']]
predict_request = np.array(predict_request)
# Predict using the random forest model
y = random_forest_model.predict(predict_request)
# Return prediction
output = [y[0]]
return jsonify(results=output)
if __name__ == '__main__':
app.run(port = 9000, debug = True)