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):
|