File size: 470 Bytes
63ef347 | 1 2 3 4 5 6 7 | from pathlib import Path
import sys,numpy as np
R=Path(__file__).resolve().parents[1];sys.path.insert(0,str(R));from model.climatenet import *
c=cfg(R);x=[];y=[]
for i in range(24):a,b=sample(i);x.append(a);y.append(b)
p=R/c['data']['path'];p.parent.mkdir(parents=True,exist_ok=True);np.savez_compressed(p,input=x,target=y,split=[0]*16+[1]*4+[2]*4,origins=np.array([[i*20%1056,i*25%672] for i in range(24)]),logical_shape=[4,1152,768],is_complete_global=False);print(p)
|