From f720529ac9a3da70163c406fef27c9882b154d01 Mon Sep 17 00:00:00 2001
From: Chintan Shah <chintan.shah@apprenticehealth.com>
Date: Tue, 8 Oct 2019 13:11:43 -0400
Subject: [PATCH] demoing with test data

---
 run_demo_pytorch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/run_demo_pytorch.py b/run_demo_pytorch.py
index 7e27667..693dfad 100644
--- a/run_demo_pytorch.py
+++ b/run_demo_pytorch.py
@@ -19,7 +19,7 @@ def run_dcrnn(args):
         #     tf_config = tf.ConfigProto(device_count={'GPU': 0})
         # with tf.Session(config=tf_config) as sess:
         supervisor = DCRNNSupervisor(adj_mx=adj_mx, **supervisor_config)
-        mean_score, outputs = supervisor.evaluate()
+        mean_score, outputs = supervisor.evaluate('test')
         np.savez_compressed(args.output_filename, **outputs)
         print('Predictions saved as {}.'.format(args.output_filename))