Skip to content

Instantly share code, notes, and snippets.

@bgweber
Created April 5, 2020 18:49
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save bgweber/1db22cf2270f014fa7d7db5507544bd4 to your computer and use it in GitHub Desktop.
Save bgweber/1db22cf2270f014fa7d7db5507544bd4 to your computer and use it in GitHub Desktop.
import pandas as pd
from sklearn.linear_model import LogisticRegression
import flask
df = pd.read_csv("https://github.com/bgweber/Twitch/raw/master/Recommendations/games-expand.csv")
model = LogisticRegression()
model.fit(df.drop(['label'], axis=1), df['label'])
app = flask.Flask(__name__)
@app.route("/", methods=["GET","POST"])
def predict():
data = {"success": False}
params = flask.request.json
if params is None:
params = flask.request.args
if "G1" in params.keys():
new_row = { "G1": params.get("G1"), "G2": params.get("G2"),
"G3": params.get("G3"), "G4": params.get("G4"),
"G5": params.get("G5"), "G6": params.get("G6"),
"G7": params.get("G7"), "G8": params.get("G8"),
"G9": params.get("G9"), "G10": params.get("G10") }
new_x = pd.DataFrame.from_dict(new_row, orient = "index").transpose()
data["response"] = str(model.predict_proba(new_x)[0][1])
data["success"] = True
return flask.jsonify(data)
if __name__ == '__main__':
app.run(host='0.0.0.0', port=80)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment