| from pathlib import Path | |
| import sys,numpy as np | |
| ROOT=Path(__file__).resolve().parents[1];sys.path.insert(0,str(ROOT)) | |
| from model.ace2_seasonal import * | |
| c=load_config(ROOT);x=np.stack([state(i).numpy() for i in range(8)]);y=np.stack([state(i,1).numpy() for i in range(8)]);path=ROOT/c["data"]["path"];path.parent.mkdir(parents=True,exist_ok=True);np.savez_compressed(path,input=x,target=y,origins=np.array([[20,40],[40,80],[60,120],[80,160],[100,200],[120,240],[140,280],[150,320]]),logical_shape=np.array([8,180,360]),is_complete_global=False);print(path) | |