Merge pull request #5 from fullstack-ml-academy/dig-5-train-model-github-action

Train Model Github Action

Author: LauraBoenchenLB

.github/workflows/model-training.yml

@@ -0,0 +1,20 @@
+name: model-training
+on: [push]
+jobs:
+  train-model:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout Repository
+        uses: actions/checkout@v3
+      - name: Setup Python und install requirements
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.8"
+      - run: pip install -r requirements.txt
+      - name: Train model
+        run: python src/train.py
+      - name: Upload trained model
+        uses: actions/upload-artifact@v4
+        with:
+          name: regressor_mpg.pickle
+          path: data/models/regressor_mpg.pickle

requirements.txt

@@ -7,3 +7,5 @@ scikit-learn==1.0.2
 scipy==1.8.0
 six==1.16.0
 threadpoolctl==3.1.0
+flask==3.1.1
+flask_cors==6.0.1

src/api.py

@@ -0,0 +1,32 @@
+from flask import Flask, Response, request
+from flask_cors import CORS
+import os
+import pandas as pd
+import pickle
+
+app = Flask(__name__)
+
+CORS(app)
+
+# load training data
+training_data = pd.read_csv(os.path.join("data", "auto-mpg.csv"))
+
+# load model
+file_to_open = open(os.path.join("data", "models", "baummethoden_lr.pickle"), "rb")
+trained_model = pickle.load(file_to_open)
+file_to_open.close()
+
+
+@app.route("/", methods=["GET"])
+def index():
+    return {"hello": "world"}
+
+
+@app.route("/hello_world", methods=["GET"])
+def hello_world():
+    return "<p>Hello World!</p>"
+
+
+@app.route("/training_data", methods=["GET"])
+def get_training_data():
+    return Response(training_data.to_json(), mimetype="application/json")

wsgi.py

@@ -0,0 +1,5 @@
+from src.api import app
+
+
+if __name__ == "__main__":
+    app.run(debug=True)