Update trainer.py

trainer.py

@@ -915,10 +915,10 @@ class Trainer(object):
         
         # print("Metrics calculation started using MOSES.")
         return {
-            "runtime": et,
-            "fraction_valid": fraction_valid(metric_calc_dr),
-            "uniqueness": fraction_unique(metric_calc_dr),
-            "novelty": novelty(metric_calc_dr, drug_smiles),
+            "Runtime (seconds)": round(et, 2),
+            "Validity": f"{fraction_valid(metric_calc_dr)*100:.2f}%",
+            "Uniqueness": f"{fraction_unique(metric_calc_dr)*100:.2f}%",
+            "Novelty": f"{novelty(metric_calc_dr, drug_smiles)*100:.2f}%",
         }
         # print("Validity: ", fraction_valid(metric_calc_dr), "\n")
         # print("Uniqueness: ", fraction_unique(metric_calc_dr), "\n")