ium_495719/.github/workflows/main.yml

79 lines
1.8 KiB
YAML
Raw Normal View History

2024-06-06 02:12:40 +02:00
name: Housing Price Workflow
2024-06-06 01:47:20 +02:00
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:
2024-06-06 02:23:23 +02:00
train:
2024-06-06 01:47:20 +02:00
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
2024-06-06 01:59:07 +02:00
run: python ./github_project/create_model.py ${{ github.event.inputs.epochs }} ${{ github.event.inputs.learning_rate }} ${{ github.event.inputs.batch_size }}
2024-06-06 01:47:20 +02:00
2024-06-06 02:29:50 +02:00
- name: Upload Artefacts
uses: actions/upload-artifact@v2
with:
name: model
path: |
./github_project/hp_model.h5
2024-06-06 02:23:23 +02:00
evaluate:
needs: train
2024-06-06 02:18:13 +02:00
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'
2024-06-06 02:23:23 +02:00
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install pandas scikit-learn tensorflow matplotlib mlflow
2024-06-06 01:47:20 +02:00
- name: Evaluate Model
2024-06-06 02:18:13 +02:00
run: python ./github_project/evaluate.py ${{ github.run_number }}
2024-06-06 02:29:50 +02:00
- name: Upload Artefacts
uses: actions/upload-artifact@v2
with:
name: evaluation-results
path: |
./github_project/hp_test_predictions.csv
./github_project/hp_test_metrics.csv
plot_mae.png
plot_r2.png
plot_rmse.png