train.yml

name: Train and Save Models and Hyperparameters

on:
  push:
    branches: [ main ]
  pull_request:
    branches: [ main ]

jobs:
  train-and-save:
    runs-on: ubuntu-latest
    steps:
      - name: Checkout code
        uses: actions/checkout@v4

      - name: Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '3.12'

      - name: Install dependencies
        run: |
          python -m pip install --upgrade pip
          pip install -r requirements.txt

      - name: Train and save all models
        run: python train.py

      - name: Upload trained models and params
        uses: actions/upload-artifact@v4
        with:
          name: trained-models
          path: |
            models/*.joblib
            data/processed/*.json