geolvr / ADSNet

Code release for "Attention-Based Dual-Source Spatiotemporal Neural Network for Lightning Forecast"

ADSNet

This is a Keras implementation of ADSNet, a spatiotemporal neural network for 12-hour grid-level lightning forecast.

core.py ---- For training and inference
generator.py ---- Data loader
new_models.py ---- Define models
scores.py ---- For calculating evaluation metrics

Requirements:
Keras 2.2.4 + Numpy