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

Notebooks/analysis #121

Closed tommylees112 closed 4 years ago

tommylees112 commented 5 years ago

Producing the final notebooks as part of the ECMWF Summer of Weather Code.

Split into just two notebooks: 1) Pipeline.ipynb

2) Analysis.ipynb

Requires:

  1. A zip file (data/zip_data.zip)
  2. How to share this data? Can we use GoogleDrive? Too big too share via github unfortunately (~1GB)
  3. We need the trained model.pt file (my one doesn't seem to produce our sota RMSE (i get an rmse ~40)
  4. need help with the shap values. need to be computed so that can be plotted (using the static cell from @gabrieltseng's draft notebook
gabrieltseng commented 4 years ago

Nice work Tommy.

Some of the notebooks seem incomplete / don't run? (Specifically, 18_tl_analysis_chirps_grid.ipynb). It's probably fine, since its in the drafts, but is this a bug in code which should be fixed?

tommylees112 commented 4 years ago

okay i think i fixed that bug!

gabrieltseng commented 4 years ago

Nice ! Could you rerun that notebook too (just prevents buggy notebooks from being uploaded)

gabrieltseng commented 4 years ago

LGTM once that happens