just display important params

EleutherAI · Aug 22, 2023 · 10e4b9b · 10e4b9b
1 parent 10b8973
commit 10e4b9b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/comp.py b/comp.py
@@ -16,6 +16,12 @@
 # reinsert the non-float columns at the front
 df_diff = pd.concat([df.select_dtypes(include=["object", "int"]), df_diff], axis=1)
 
+# only include the columns dataset layer ensembling auroc_estimate acc_estimate
+df_diff = df_diff.filter(
+    ["dataset", "layer", "ensembling", "auroc_estimate", "acc_estimate"]
+)
+
+
 import rich
 
 rich.print(df_diff)