Spaces:
Running
Running
Commit
β’
e89d633
1
Parent(s):
7c4e3f0
fix bench_df modifications
Browse files
app.py
CHANGED
@@ -120,24 +120,26 @@ def get_benchmark_df(benchmark="1xA100-80GB"):
|
|
120 |
|
121 |
def get_benchmark_table(bench_df):
|
122 |
# add * to quantized models score
|
123 |
-
|
|
|
|
|
124 |
lambda x: f"{x['best_score']}**" if x["quantized"] else x["best_score"],
|
125 |
axis=1,
|
126 |
)
|
127 |
|
128 |
# sort
|
129 |
-
|
130 |
# filter
|
131 |
-
|
132 |
# rename
|
133 |
-
|
134 |
# transform
|
135 |
-
|
136 |
-
|
137 |
process_model_name
|
138 |
)
|
139 |
|
140 |
-
return
|
141 |
|
142 |
|
143 |
def get_benchmark_plot(bench_df):
|
|
|
120 |
|
121 |
def get_benchmark_table(bench_df):
|
122 |
# add * to quantized models score
|
123 |
+
copy_df = bench_df.copy()
|
124 |
+
|
125 |
+
copy_df["best_score"] = copy_df.apply(
|
126 |
lambda x: f"{x['best_score']}**" if x["quantized"] else x["best_score"],
|
127 |
axis=1,
|
128 |
)
|
129 |
|
130 |
# sort
|
131 |
+
copy_df.sort_values(by=SORTING_COLUMN, ascending=True, inplace=True)
|
132 |
# filter
|
133 |
+
copy_df = copy_df[list(ALL_COLUMNS_MAPPING.keys())]
|
134 |
# rename
|
135 |
+
copy_df.rename(columns=ALL_COLUMNS_MAPPING, inplace=True)
|
136 |
# transform
|
137 |
+
copy_df["Type π€"] = copy_df["Type π€"].apply(process_model_type)
|
138 |
+
copy_df["Best Scored Model π"] = copy_df["Best Scored Model π"].apply(
|
139 |
process_model_name
|
140 |
)
|
141 |
|
142 |
+
return copy_df
|
143 |
|
144 |
|
145 |
def get_benchmark_plot(bench_df):
|