This repository contains various configurations to guide users through a full machine learning pipeline for weather prediction! You will find multiple directories showcasing various model configurations ranging from a "hello world" setup to operational quality models. While various model configurations will differ, they will all follow the same steps of an end-to-end machine learning pipeline. This includes:
- Proper environment setup: Users are provided with ready-to-use conda environments
- Prepare training data: Use
ufs2arcoto create training, validation, and test datasets - Train an AI model: use
anemoi-coremodules to train a graph-based model for weather prediction - Generate a forecast: use
anemoi-inferenceto run inference from a model checkpoint - Verify model performance: use
wxvxto verify forecasts against gridded analysis or observervations - View model output: use
eagle-toolsto visualize model output and scores
For more information about model configurations or the various steps of the pipeline, please see our documentation.
ufs2arco: Tim Smith (NOAA Physical Sciences Laboratory)
Anemoi: European Centre for Medium-Range Weather Forecasts
- anemoi-core github
- anemoi-inference github
- Documentation: anemoi-models, anemoi-graphs, anemoi-training, anemoi-inference
wxvx: Paul Madden (NOAA Global Systems Laboratory/Cooperative Institute for Research In Environmental Sciences)
eagle-tools: Tim Smith (NOAA Physical Sciences Laboratory)