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))