-
Notifications
You must be signed in to change notification settings - Fork 2
/
server.py
26 lines (19 loc) · 666 Bytes
/
server.py
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
from flask import Flask, jsonify, request
import pandas as pd
import joblib
app = Flask(__name__)
@app.route("/predict", methods=['POST'])
def do_prediction():
json = request.get_json()
model = joblib.load('model/rf_model.pkl')
df = pd.DataFrame(json, index=[0])
from sklearn.preprocessing import StandardScaler
scaler = StandardScaler()
scaler.fit(df)
df_x_scaled = scaler.transform(df)
df_x_scaled = pd.DataFrame(df_x_scaled, columns=df.columns)
y_predict = model.predict(df_x_scaled)
result = {"Predicted House Price" : y_predict[0]}
return jsonify(result)
if __name__ == "__main__":
app.run(host='0.0.0.0')