Upload dedupe.py

chess-gpt-eval/dedupe.py

@@ -0,0 +1,26 @@
+import pandas as pd
+
+filename = 'logs/Mamba/6_6M/1way_ckpt_2741760b_pt_vs_lc0_sweep.csv'
+#filename = 'logs/Mamba/11M/ckpt_1188012b_pt_vs_lc0_sweep.csv'
+#filename = 'logs/11M/Round 1/ckpt_2608480_pt_vs_lc0_sweep.csv'
+
+# Read in the CSV file
+df = pd.read_csv(filename)
+
+# Count the original total
+original_total = df.shape[0]
+
+# Filter out duplicates in the 'transcript' column
+df = df.drop_duplicates(subset='transcript')
+
+# Count the remove, and remaining rows
+removed = original_total - df.shape[0]
+remaining = df.shape[0]
+
+# Print out the results
+print("Original total rows:", original_total)
+print("Removed rows:", removed)
+print("Remaining rows:", remaining)
+
+# Write the filtered data to a new CSV file
+df.to_csv(filename, index=False)