| from pathlib import Path | |
| import sys,numpy as np,matplotlib;matplotlib.use('Agg');import matplotlib.pyplot as plt;R=Path(__file__).resolve().parents[1];sys.path.insert(0,str(R));from model.rainrunoff_lstm import * | |
| c=cfg(R);d=np.load(R/c['paths']['predictions']);nse=1-((d['prediction']-d['target'])**2).sum()/((d['target']-d['target'].mean())**2).sum();write(R/c['paths']['evaluation'],{'nse':float(nse)});plt.plot(d['target']);plt.plot(d['prediction']);p=R/c['paths']['figure'];p.parent.mkdir(parents=True,exist_ok=True);plt.savefig(p);print(p) | |