FilesExpand file tree

main

/

summarize_results.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

25 lines (18 loc) · 476 Bytes

main

/

summarize_results.py

Copy path

Top

File metadata and controls

• Code
• Blame

25 lines (18 loc) · 476 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

import pandas as pd

RESULTS_PATH = 'all_results_20250930-154315.csv'

PREFIXES = [

'all',

'truncated',

'non_truncated',

]

cols = [

'test rmse',

# 'test rmse_pcutoff',

# 'test mAP',

# 'test mAR',

]

results_df = pd.read_csv(RESULTS_PATH)

# group by experiment_id and average

grouped = results_df.groupby('experiment').mean(numeric_only=True)

select_cols = [f'{prefix}__{col}' for prefix in PREFIXES for col in cols]

print(grouped[select_cols])