Skip to content

EyringMLClimateGroup/heuer23_ml_convection_parameterization

Repository files navigation

Interpretable multiscale Machine Learning-Based Parameterizations of Convection for ICON

This repository contains the code for the developement of data-driven convection parameterizations based on the NARVAL data set for ICON-A

The corresponding paper is available as a preprint on arXiv

Heuer, Helge, et al. "Interpretable multiscale Machine Learning-Based Parameterizations of Convection for ICON." arXiv preprint arXiv:2311.03251 (2023). https://doi.org/10.48550/arXiv.2311.03251

Corresponding DOI: DOI

If you want to use this repository, start by executing

pip install -e .

this will make it possible to import various python functions in jupyter notebooks inside this repository.

Repository content

Data

To fully reproduce the results it is first necessary to have access to accounts on DKRZ/Levante and the narval simulations. The coarse graining and preprocessing scripts are found in preprocessing. For training and evaluation of the trained networks, some sample data has been saved in the preprocessed data directory.

Dependencies

  • Xarray
  • Numba
  • Pytorch
  • Scikit-learn
  • Ray
  • Dask
  • Netcdf4
  • Other packages like numpy, matplotlib, pandas, joblib, and tqdm