Skip to content

Commit c155467

Browse files
committed
Add script to list all metrics per dataset
1 parent 4fef880 commit c155467

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

normalize_metrics.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
#!/usr/bin/env python
2+
3+
import fire
4+
from label_tables import get_sota_tasks
5+
6+
7+
def normalize_metrics(tasksfile):
8+
tasks = get_sota_tasks(tasksfile)
9+
10+
print("task\tdataset\tmetric")
11+
for task in tasks:
12+
for dataset in task.datasets:
13+
for row in dataset.sota.rows:
14+
for metric in row.metrics:
15+
print(f"{task.name}\t{dataset.name}\t{metric}")
16+
17+
18+
if __name__ == "__main__": fire.Fire(normalize_metrics)

0 commit comments

Comments
 (0)