Add IFS and GFS support

secondlaw-ai / skyrim

🌎 🤝 AI weather models united

Apache License 2.0

158 stars 7 forks source link

Add IFS and GFS support #15

Closed m13uz closed 5 months ago

m13uz commented 5 months ago

Fixed ifs initial conditions (start time should be set properly though, i.e. 00, 06, 12, 18)
Implemented nwp/ifs.py

Now, we can get IFS/HRES predictions as follows:

from skyrim.libs.nwp import IFSModel
model = IFSModel(channels=["u10m", "v10m", "t2m"])
pred = model.predict(
    date="20240601",
    time="0000",
    lead_time=12,
    save=True
)
pred.shape
>> (5, 3, 721, 1440)

m13uz commented 5 months ago

will add more test with additional PR