saicharan2804 commited on
Commit
2a3cfbf
·
1 Parent(s): d84e30a

Removed some metrics

Browse files
Files changed (1) hide show
  1. molgenevalmetric.py +4 -1
molgenevalmetric.py CHANGED
@@ -131,6 +131,9 @@ class molgenevalmetric(evaluate.Metric):
131
  score = sum(score) / len(score)
132
 
133
  Results.update({f"{oracle_name}": score})
134
-
 
 
 
135
 
136
  return {"results": Results}
 
131
  score = sum(score) / len(score)
132
 
133
  Results.update({f"{oracle_name}": score})
134
+
135
+ keys_to_remove = ["FCD/TestSF", "SNN/TestSF", "Frag/TestSF", "Scaf/TestSF"]
136
+ for key in keys_to_remove:
137
+ result['results'].pop(key, None)
138
 
139
  return {"results": Results}