Add github workflow
This commit is contained in:
parent
b586cbdbc4
commit
d051818515
41
.github/workflows/main.yml
vendored
Normal file
41
.github/workflows/main.yml
vendored
Normal 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 }}
|
Loading…
Reference in New Issue
Block a user