From b2d2b21dbd16b4d4454b9714ac1e2cbb1c710b3f Mon Sep 17 00:00:00 2001 From: Chintan Shah Date: Tue, 8 Oct 2019 13:46:34 -0400 Subject: [PATCH] logging MAE --- run_demo_pytorch.py | 1 + 1 file changed, 1 insertion(+) diff --git a/run_demo_pytorch.py b/run_demo_pytorch.py index 693dfad..caa9dbe 100644 --- a/run_demo_pytorch.py +++ b/run_demo_pytorch.py @@ -21,6 +21,7 @@ def run_dcrnn(args): supervisor = DCRNNSupervisor(adj_mx=adj_mx, **supervisor_config) mean_score, outputs = supervisor.evaluate('test') np.savez_compressed(args.output_filename, **outputs) + print("MAE : {}".format(mean_score)) print('Predictions saved as {}.'.format(args.output_filename))