Update trainer.py

trainer.py

@@ -917,8 +917,8 @@ class Trainer(object):
             "Runtime (seconds)": round(et, 2),
             "Validity": f"{fraction_valid(metric_calc_dr)*100:.2f}%",
             "Uniqueness": f"{fraction_unique(metric_calc_dr)*100:.2f}%",
-            "Novelty Train": f"{novelty(metric_calc_dr, smiles_train)*100:.2f}%",
-            "Novelty Test": f"{novelty(metric_calc_dr, smiles_test)*100:.2f}%"
+            "Novelty (Train)": f"{novelty(metric_calc_dr, smiles_train)*100:.2f}%",
+            "Novelty (Test)": f"{novelty(metric_calc_dr, smiles_test)*100:.2f}%"
         }
         # print("Validity: ", fraction_valid(metric_calc_dr), "\n")
         # print("Uniqueness: ", fraction_unique(metric_calc_dr), "\n")