| 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) | |