lixuejing
commited on
Commit
·
59a1e42
1
Parent(s):
447637d
add log
Browse files- 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)
|