This repository contains the machine learning code used in the paper: Multi-source seamless lightning nowcasting with recurrent deep learning, accepted to Artificial Intelligence for the Earth Systems. Preprint available: https://arxiv.org/abs/2203.10114
You need NumPy, Scipy, Matplotlib, Tensorflow, Numba, Dask and NetCDF4 for Python.
Clone the repository, then, in the main directory, run
$ python setup.py develop(if you plan to modify the code) or
$ python setup.py installif you just want to run it.
The dataset can be found at the following Zenodo repository: https://doi.org/10.5281/zenodo.6325370
Download the NetCDF file. You can place it in the data directory but elsewhere on the disk works too.
Go to the scripts directory and start an interactive shell. There, you can find training.py that contains the script you need for training and plots_lightning.py that produces the plots from the paper.