| from pathlib import Path | |
| import sys,numpy as np | |
| ROOT=Path(__file__).resolve().parents[1];sys.path.insert(0,str(ROOT)) | |
| from model.weatherbench import * | |
| c=load_config(ROOT);x=np.stack([synthetic_state(i).numpy() for i in range(24)]);y6=np.stack([synthetic_state(i,1).numpy() for i in range(24)]);y3=np.stack([synthetic_state(i,12).numpy() for i in range(24)]);y5=np.stack([synthetic_state(i,20).numpy() for i in range(24)]);split=np.array([0]*16+[1]*4+[2]*4);path=ROOT/c["data"]["path"];path.parent.mkdir(parents=True,exist_ok=True);np.savez_compressed(path,input=x,target_6h=y6,target_3d=y3,target_5d=y5,split=split,latitude=np.linspace(-90,90,32),longitude=np.linspace(0,360,64,endpoint=False),pressure_levels=np.array(PRESSURE_LEVELS));print(path) | |