From 9216ba8b13105298bcb5ec77825d390ef961a82f Mon Sep 17 00:00:00 2001
From: rabbit721 <001207jiaojie@gmail.com>
Date: Sat, 10 Oct 2020 23:23:25 -0400
Subject: [PATCH] added printing for pred_error at test time

---
 model_arch.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/model_arch.py b/model_arch.py
index f360248..d194507 100644
--- a/model_arch.py
+++ b/model_arch.py
@@ -307,8 +307,9 @@ def _forward(self, epoch):
             self.weighted_mae = total_mean_mae / data_size
 
             if epoch % 50 == 0:
-                print("test batch R(q): {}, Accumulated Error: "\
-                      "{}, Weighted MAE: {}".format(self.rq,
+                print("test batch Pred Err: {}, R(q): {}, Accumulated Error: "\
+                      "{}, Weighted MAE: {}".format(self.pred_err, 
+                                                    self.rq,
                                                     self.accumulate_err,
                                                     self.weighted_mae))