added printing for pred_error at test time

cmu-db · Oct 11, 2020 · 9216ba8 · 9216ba8
1 parent 3c48eac
commit 9216ba8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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))