parse_sweep.py

#!/usr/bin/env python3
import argparse
import re
import csv

def main(input_file, output_file):
    with open(input_file, 'r') as f:
        content = f.read()

    # Parse each block together to ensure values stay aligned
    pattern = re.compile(
        r'Alpha: ([\d.]+)\s+'
        r'Performance: ([\d.]+)\s+'
        r'Total Cost: ([\d.]+)',
        re.MULTILINE
    )

    rows = pattern.findall(content)

    # Write to CSV
    with open(output_file, 'w', newline='') as f:
        writer = csv.writer(f)
        writer.writerow(['alpha', 'performance', 'cost'])
        writer.writerows(rows)

    print(f"Wrote {len(rows)} rows to {output_file}")

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("--input-file", help="Input file")
    parser.add_argument("--output-file", help="Output file")
    args = parser.parse_args()
    main(args.input_file, args.output_file)