mgyigit commited on
Commit
f72f4a3
1 Parent(s): 0c5ec43

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +5 -5
inference.py CHANGED
@@ -223,11 +223,11 @@ class Inference(object):
223
  # post-process * to Carbon atom in valid molecules
224
 
225
  return{
226
- "Runtime (seconds)": f"{et}:.2f",
227
- "Validity": f"{fraction_valid(metric_calc_dr)}:.2f",
228
- "Uniqueness": f"{fraction_unique(uniqueness_calc)}:.2f",
229
- "Novelty (Train)": f"{novelty(metric_calc_dr, chembl_smiles)}:.2f",
230
- "Novelty (Inference)": f"{novelty(metric_calc_dr, chembl_test)}:.2f",
231
  }
232
 
233
 
 
223
  # post-process * to Carbon atom in valid molecules
224
 
225
  return{
226
+ "Runtime (seconds)": f"{et:.2f}",
227
+ "Validity": f"{fraction_valid(metric_calc_dr):.2f}",
228
+ "Uniqueness": f"{fraction_unique(uniqueness_calc):.2f}",
229
+ "Novelty (Train)": f"{novelty(metric_calc_dr, chembl_smiles):.2f}",
230
+ "Novelty (Inference)": f"{novelty(metric_calc_dr, chembl_test):.2f}",
231
  }
232
 
233