Spaces:
Running
Running
Update utils_v2.py
Browse files- utils_v2.py +2 -3
utils_v2.py
CHANGED
|
@@ -93,10 +93,9 @@ def load_scores(raw_scores=None):
|
|
| 93 |
score = raw_scores.get(modality, {}).get(dataset, 0.0)
|
| 94 |
score = 0.0 if score == "FILE_N/A" else score
|
| 95 |
metric = SPECIAL_METRICS.get(dataset, 'hit@1')
|
| 96 |
-
# metric = "ndcg@5" if modality == 'visdoc' else metric
|
| 97 |
-
if modality == 'visdoc':
|
| 98 |
-
metric = "ndcg_linear@5" if "ndcg_linear@5" in score else "ndcg@5"
|
| 99 |
if isinstance(score, dict):
|
|
|
|
|
|
|
| 100 |
score = score.get(metric, 0.0)
|
| 101 |
all_scores[dataset] = round(score, 2)
|
| 102 |
return all_scores
|
|
|
|
| 93 |
score = raw_scores.get(modality, {}).get(dataset, 0.0)
|
| 94 |
score = 0.0 if score == "FILE_N/A" else score
|
| 95 |
metric = SPECIAL_METRICS.get(dataset, 'hit@1')
|
|
|
|
|
|
|
|
|
|
| 96 |
if isinstance(score, dict):
|
| 97 |
+
if modality == 'visdoc':
|
| 98 |
+
metric = "ndcg_linear@5" if "ndcg_linear@5" in score else "ndcg@5"
|
| 99 |
score = score.get(metric, 0.0)
|
| 100 |
all_scores[dataset] = round(score, 2)
|
| 101 |
return all_scores
|