openclimatefix / graph_weather

PyTorch implementation of Ryan Keisler's 2022 "Forecasting Global Weather with Graph Neural Networks" paper (https://arxiv.org/abs/2202.07575)
MIT License
187 stars 47 forks source link

Add IFS Analysis dataloader for training #95

Closed 0xFrama closed 5 months ago

0xFrama commented 6 months ago

Pull Request

Description

This PR adds a dataloader for the IFS Analysis dataset (for training). Each input/output data has the following size: [2048, 78].

This is an incomplete PR because what is missing is the second dataset (IFS HRES/ENS) for evaluation. I'm still in the process of calculating the mean and std for the latter.

I made this PR to show whether what has been done so far is fine or it needs some adjustment.

If this has to be a complete PR, you can close it and I can make a new one with what is missing.

Fixes #86

Checklist:

jacobbieker commented 5 months ago

Sorry, again missed the notification, this looks good! I'll merge it.