czzhangheng
/
DCRNN
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

logging MAE

This commit is contained in:
Chintan Shah 2019-10-08 13:46:34 -04:00
parent f720529ac9
commit b2d2b21dbd
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
run_demo_pytorch.py
View File

@ -21,6 +21,7 @@ def run_dcrnn(args):
supervisor = DCRNNSupervisor(adj_mx=adj_mx, **supervisor_config) supervisor = DCRNNSupervisor(adj_mx=adj_mx, **supervisor_config)
mean_score, outputs = supervisor.evaluate('test') mean_score, outputs = supervisor.evaluate('test')
np.savez_compressed(args.output_filename, **outputs) np.savez_compressed(args.output_filename, **outputs)
print("MAE : {}".format(mean_score))
print('Predictions saved as {}.'.format(args.output_filename)) print('Predictions saved as {}.'.format(args.output_filename))