← All projects
Machine Learning Beginner

StormCast

Thunderstorm forecasting with MLflow tracking

Forecast thunderstorm risk from weather-station data and track every experiment, parameter and model version with MLflow.

SHIPSTACK · PROJECT StormCast Thunderstorm forecasting withMLflow tracking Beginner Machine Learning 1 Station data 2 Features 3 MLflow 4 Forecast API 5 milestones

The scenario

StormCast teaches disciplined experimentation: time-aware train/test splits, feature lags, several model families, and MLflow tracking and model registry so you always know which model is in production and why.

System architecture

01 Station data Hourly readings
02 Features Lags + rolling stats
03 MLflow Experiments + registry
04 Forecast API Storm risk

High-level flow of StormCast, from input to output.

What you will learn

  • Time-series aware validation
  • Experiment tracking
  • Model registry and promotion

What you will build

  • Feature pipeline
  • Tracked experiments
  • Forecast API

Build plan

  1. Data
  2. Features
  3. Experiments
  4. Registry
  5. API

Tech stack

pandasscikit-learnMLflowFastAPI