lixuejing commited on
Commit
59a1e42
·
1 Parent(s): 447637d
Files changed (1) hide show
  1. src/tools/plots.py +2 -1
src/tools/plots.py CHANGED
@@ -39,6 +39,7 @@ def create_scores_df(raw_data: list[EvalResult]) -> pd.DataFrame:
39
  current_date = row["date"]
40
  if task.benchmark == "Average":
41
  current_score = np.mean(list(row["results"].values()))
 
42
  else:
43
  if row["results"]:
44
  current_score = row["results"][task.benchmark]
@@ -70,10 +71,10 @@ def create_plot_df(scores_df: dict[str: pd.DataFrame]) -> pd.DataFrame:
70
  # Iterate over the cols and create a new DataFrame for each column
71
  for col in BENCHMARK_COLS + [AutoEvalColumn.average.name]:
72
  print("col",col)
73
- print("dfs", dfs)
74
  d = scores_df[col].reset_index(drop=True)
75
  d["task"] = col
76
  dfs.append(d)
 
77
 
78
  # Concatenate all the created DataFrames
79
  concat_df = pd.concat(dfs, ignore_index=True)
 
39
  current_date = row["date"]
40
  if task.benchmark == "Average":
41
  current_score = np.mean(list(row["results"].values()))
42
+ print("average", current_score)
43
  else:
44
  if row["results"]:
45
  current_score = row["results"][task.benchmark]
 
71
  # Iterate over the cols and create a new DataFrame for each column
72
  for col in BENCHMARK_COLS + [AutoEvalColumn.average.name]:
73
  print("col",col)
 
74
  d = scores_df[col].reset_index(drop=True)
75
  d["task"] = col
76
  dfs.append(d)
77
+ print("dfs", dfs)
78
 
79
  # Concatenate all the created DataFrames
80
  concat_df = pd.concat(dfs, ignore_index=True)