Skip to content

Commit 2071b10

Browse files
committed
Fix GMTKN55 metric order
1 parent 6657747 commit 2071b10

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

ml_peg/analysis/molecular/GMTKN55/analyse_GMTKN55.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff 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}

0 commit comments

Comments
 (0)