From d05181851517d62c9ee6b20a2a7383073fb03c28 Mon Sep 17 00:00:00 2001
From: PawelDopierala <pawel.dopierala4@gmail.com>
Date: Thu, 6 Jun 2024 01:47:20 +0200
Subject: [PATCH] Add github workflow

---
 .github/workflows/main.yml | 41 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 .github/workflows/main.yml

diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
new file mode 100644
index 0000000..71ff769
--- /dev/null
+++ b/.github/workflows/main.yml
@@ -0,0 +1,41 @@
+name: ML Workflow
+
+on:
+  workflow_dispatch:
+    inputs:
+      epochs:
+        description: 'Number of epochs'
+        required: true
+        default: 20
+      learning_rate:
+        description: 'Learning rate'
+        required: true
+        default: 0.001
+      batch_size:
+        description: 'Batch size'
+        required: true
+        default: 32
+
+jobs:
+  train_and_evaluate:
+    runs-on: ubuntu-latest
+
+    steps:
+    - name: Checkout repository
+      uses: actions/checkout@v2
+
+    - name: Set up Python
+      uses: actions/setup-python@v2
+      with:
+        python-version: '3.8'
+
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install pandas scikit-learn tensorflow matplotlib mlflow
+
+    - name: Train Model
+      run: python create_model.py ${{ github.event.inputs.epochs }} ${{ github.event.inputs.learning_rate }} ${{ github.event.inputs.batch_size }}
+
+    - name: Evaluate Model
+      run: python evaluate.py ${{ github.run_number }}
\ No newline at end of file