ECMWFCode4Earth / ml_drought

Machine learning to better predict and understand drought. Moving github.com/ml-clim
https://ml-clim.github.io/drought-prediction/
90 stars 18 forks source link

Add EALSTM #76

Closed gabrieltseng closed 5 years ago

gabrieltseng commented 5 years ago

Adds an entity-aware LSTM, as described here: https://github.com/kratzert/ealstm_regional_modeling

Most of the code is just copied from the EALSTM described there, and fitted to our current RNN model.

Currently, the static data is the latlon data, but this should be updated to include landcover, and eventually topographical data.

gabrieltseng commented 5 years ago

TODO: add mean values

gabrieltseng commented 5 years ago

Added mean values. Specifically:

gabrieltseng commented 5 years ago

Added standard deviations too