Spaces:
Running
Running
Update inference.py
Browse files- 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
|
227 |
-
"Validity": f"{fraction_valid(metric_calc_dr)
|
228 |
-
"Uniqueness": f"{fraction_unique(uniqueness_calc)
|
229 |
-
"Novelty (Train)": f"{novelty(metric_calc_dr, chembl_smiles)
|
230 |
-
"Novelty (Inference)": f"{novelty(metric_calc_dr, chembl_test)
|
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 |
|