Commit 122cab3c 122cab3c2eecc8ca0c8464e3b5a6dd4ce45b9ce2 by 沈秋雨

优化fix脚本

1 parent d0592b22
......@@ -229,10 +229,12 @@ def main():
print(f"移除行数: {len(removed)}(同歌手且时长相近)")
print(f"保留行数: {len(kept)}")
# 写修复后的 CSV
# 写修复后的 CSV(去掉冗余列)
DROP_COLS = {"topk_hit", "expected_duplicate", "predicted_duplicate", "sample_class"}
out_fieldnames = [f for f in fieldnames if f not in DROP_COLS]
out_path = Path(args.output)
with out_path.open("w", newline="", encoding="utf-8") as f:
writer = csv.DictWriter(f, fieldnames=fieldnames)
writer = csv.DictWriter(f, fieldnames=out_fieldnames, extrasaction="ignore")
writer.writeheader()
writer.writerows(kept)
print(f"已写入: {out_path}")
......