Spaces:
Sleeping
Sleeping
Update evaluate.py
Browse files- evaluate.py +2 -1
evaluate.py
CHANGED
|
@@ -4,7 +4,8 @@ from vg_evaluator import evaluation_for_challenge
|
|
| 4 |
from datasets import load_dataset
|
| 5 |
import os
|
| 6 |
kk = load_dataset(os.getenv('VG_DATA'))['test'].to_dict()
|
| 7 |
-
|
|
|
|
| 8 |
print(len(kk))
|
| 9 |
print(kk.keys())
|
| 10 |
def run_evaluation(pred_path):
|
|
|
|
| 4 |
from datasets import load_dataset
|
| 5 |
import os
|
| 6 |
kk = load_dataset(os.getenv('VG_DATA'))['test'].to_dict()
|
| 7 |
+
for key in kk.keys():
|
| 8 |
+
kk[key] = kk[key][0]
|
| 9 |
print(len(kk))
|
| 10 |
print(kk.keys())
|
| 11 |
def run_evaluation(pred_path):
|