fix github workflow

This commit is contained in:
PawelDopierala 2024-06-06 02:29:50 +02:00
parent 0dc2c077bc
commit 6b77f00b66
1 changed files with 19 additions and 0 deletions

View File

@ -37,6 +37,13 @@ jobs:
- name: Train Model - name: Train Model
run: python ./github_project/create_model.py ${{ github.event.inputs.epochs }} ${{ github.event.inputs.learning_rate }} ${{ github.event.inputs.batch_size }} run: python ./github_project/create_model.py ${{ github.event.inputs.epochs }} ${{ github.event.inputs.learning_rate }} ${{ github.event.inputs.batch_size }}
- name: Upload Artefacts
uses: actions/upload-artifact@v2
with:
name: model
path: |
./github_project/hp_model.h5
evaluate: evaluate:
needs: train needs: train
runs-on: ubuntu-latest runs-on: ubuntu-latest
@ -57,3 +64,15 @@ jobs:
- name: Evaluate Model - name: Evaluate Model
run: python ./github_project/evaluate.py ${{ github.run_number }} run: python ./github_project/evaluate.py ${{ github.run_number }}
- 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