mgyigit commited on
Commit
8a4ac10
1 Parent(s): 9b2b625

Update inference.py

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