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
Installation
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 install
if you just want to run it.
Downloading data
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.
Running
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.