File tree Expand file tree Collapse file tree 1 file changed +3
-4
lines changed
ml_peg/analysis/molecular/GMTKN55 Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -309,13 +309,12 @@ def metrics(
309309 "Reaction barrier heights" : "Barrier heights" ,
310310 "Intramolecular noncovalent interactions" : "Intramolecular NCIs" ,
311311 "Intermolecular noncovalent interactions" : "Intermolecular NCIs" ,
312- "All (WTMAD)" : "All (WTMAD)" ,
313312 }
314313
315314 metrics = {}
316- for category in category_errors [ MODELS [ 0 ]] :
317- metrics [category_abbrevs [ category ] ] = {
318- model : category_errors [model ][category ] for model in MODELS
315+ for full_category , short_category in category_abbrevs . items () :
316+ metrics [short_category ] = {
317+ model : category_errors [model ][full_category ] for model in MODELS
319318 }
320319
321320 return metrics | {"WTMAD" : weighted_error }
You can’t perform that action at this time.
0 commit comments