Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
2836709
1
Parent(s):
5943030
Update DataFrame columns in load_retrieval_results function for clarity and accuracy
Browse files
utils.py
CHANGED
@@ -16,7 +16,7 @@ def load_retrieval_results():
|
|
16 |
|
17 |
retrieval_dataframe_path = results_dir / "retrieval_results.json"
|
18 |
if not retrieval_dataframe_path.exists():
|
19 |
-
df = pd.DataFrame(columns=["Model", "Revision", "Precision", "Task", "Model Size", "Embedding Dimension", "Max Tokens", "
|
20 |
else:
|
21 |
df = pd.read_json(retrieval_dataframe_path)
|
22 |
return df
|
|
|
16 |
|
17 |
retrieval_dataframe_path = results_dir / "retrieval_results.json"
|
18 |
if not retrieval_dataframe_path.exists():
|
19 |
+
df = pd.DataFrame(columns=["Model", "Revision", "Precision", "Task", "Model Size (in Millions)", "Embedding Dimension", "Max Tokens", "Num Likes", "Web Search Dataset (Overall Score)"])
|
20 |
else:
|
21 |
df = pd.read_json(retrieval_dataframe_path)
|
22 |
return df
|