File size: 482 Bytes
b72f6a3 | 1 2 3 4 5 | from pathlib import Path
import sys,numpy as np
R=Path(__file__).resolve().parents[1];sys.path.insert(0,str(R));from model.ai_gamfs import *
c=cfg(R);x=np.stack([state(i).numpy() for i in range(8)]);t=np.stack([[state(i,h).numpy() for h in (3,6,9,12)] for i in range(8)]);p=R/c['data']['path'];p.parent.mkdir(parents=True,exist_ok=True);np.savez_compressed(p,input=x,targets=t,leads=[3,6,9,12],aerosol_variables=AEROSOL,logical_shape=[54,361,576],is_complete_global=False);print(p)
|