ragul1995/wine

A test repository to demonstrate wandb integration with GitHub Actions for automatic metric reporting for ML models.

This repository demonstrates how to integrate GitHub Actions to:

• Upon a new commit -
  • Automatically authenticate wandb (Weights and Biases) using a custom GitHub secret.
  • Automatically train a small Random Forest Regressor model on the wine quality dataset.
  • Automatically log the training and other important model metrics to wandb.
  • Cache Python dependencies so that old dependencies do not get installed each time a run is triggered.
  • Generate a metrics.csv file after a run is successfully completed.
• Upon a new pull request -
  • Fetch the latest wandb run URL and comment that on the PR.

Workflow files

• https://github.com/sayakpaul/wine/blob/experiment-1/.github/workflows

Acknowledgments

This video by DVCorg helped me a lot to clear the initial concepts regarding GitHub Actions.