File size: 549 Bytes
48fc6c1 | 1 2 3 4 5 6 7 8 | from pathlib import Path
import sys,numpy as np,torch
R=Path(__file__).resolve().parents[1];sys.path.insert(0,str(R));from model.delta_hbv import *
c=cfg(R);fs=[];aa=[];qq=[];ee=[];teacher=DeltaHBV()
with torch.no_grad():
for i in range(12):f,a=sample(i);q,e=teacher(f[None],a[None]);fs.append(f);aa.append(a);qq.append(q[0]);ee.append(e[0])
p=R/c['data']['path'];p.parent.mkdir(parents=True,exist_ok=True);np.savez_compressed(p,forcing=np.stack(fs),attributes=np.stack(aa),streamflow=np.stack(qq),et=np.stack(ee),basin_ids=np.arange(12));print(p)
|