MohamedRashad commited on
Commit
2836709
·
1 Parent(s): 5943030

Update DataFrame columns in load_retrieval_results function for clarity and accuracy

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
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", "Web Search Dataset (MRR)", "Web Search Dataset (nDCG@k=None)"])
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