Add github workflow

This commit is contained in:
PawelDopierala 2024-06-06 01:47:20 +02:00
parent b586cbdbc4
commit d051818515

41
.github/workflows/main.yml vendored Normal file
View File

@ -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 }}