Spaces:
Sleeping
Sleeping
Commit
Β·
06b3632
1
Parent(s):
6114ec7
fix drop duplicates
Browse files
app.py
CHANGED
|
@@ -58,6 +58,15 @@ ALL_COLUMNS_DATATYPES = [
|
|
| 58 |
#
|
| 59 |
"markdown",
|
| 60 |
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 61 |
SORTING_COLUMN = ["best_score"]
|
| 62 |
SORTING_ASCENDING = [False]
|
| 63 |
|
|
@@ -99,7 +108,8 @@ def get_benchmark_df(benchmark="Succeeded-1xA100-80GB"):
|
|
| 99 |
|
| 100 |
# sort
|
| 101 |
merged_df.sort_values(by=SORTING_COLUMN, ascending=SORTING_ASCENDING, inplace=True)
|
| 102 |
-
|
|
|
|
| 103 |
return merged_df
|
| 104 |
|
| 105 |
|
|
@@ -122,8 +132,6 @@ def get_benchmark_table(bench_df):
|
|
| 122 |
copy_df["Best Scored LLM π"] = copy_df["Best Scored LLM π"].apply(
|
| 123 |
process_model_name
|
| 124 |
)
|
| 125 |
-
# drop duplicates
|
| 126 |
-
copy_df.drop_duplicates(inplace=True)
|
| 127 |
|
| 128 |
return copy_df
|
| 129 |
|
|
|
|
| 58 |
#
|
| 59 |
"markdown",
|
| 60 |
]
|
| 61 |
+
NO_DUPLICATES_COLUMNS = [
|
| 62 |
+
"backend.name",
|
| 63 |
+
"backend.torch_dtype",
|
| 64 |
+
"optimizations",
|
| 65 |
+
"quantization",
|
| 66 |
+
#
|
| 67 |
+
"weight_class",
|
| 68 |
+
"model_type",
|
| 69 |
+
]
|
| 70 |
SORTING_COLUMN = ["best_score"]
|
| 71 |
SORTING_ASCENDING = [False]
|
| 72 |
|
|
|
|
| 108 |
|
| 109 |
# sort
|
| 110 |
merged_df.sort_values(by=SORTING_COLUMN, ascending=SORTING_ASCENDING, inplace=True)
|
| 111 |
+
# drop duplicates
|
| 112 |
+
merged_df.drop_duplicates(subset=NO_DUPLICATES_COLUMNS, inplace=True)
|
| 113 |
return merged_df
|
| 114 |
|
| 115 |
|
|
|
|
| 132 |
copy_df["Best Scored LLM π"] = copy_df["Best Scored LLM π"].apply(
|
| 133 |
process_model_name
|
| 134 |
)
|
|
|
|
|
|
|
| 135 |
|
| 136 |
return copy_df
|
| 137 |
|