main.py

import yaml
import sys

from datetime import datetime
import pandas as pd

import multiprocessing as mp


from attack import run_experiments


def load_config(config_file):
    with open(config_file, 'r') as file:
        config = yaml.safe_load(file)
    return config

def main(config_file):
    config = load_config(config_file)

    results = run_experiments(config)

    results_df = pd.DataFrame(results)

    # dataFrame to CSV
    current_datetime = datetime.now()
    timestamp = current_datetime.strftime("%Y%m%d_%H%M%S")
    results_df.to_csv(f"results_{timestamp}.csv", index=False)


if __name__ == "__main__":

    mp.set_start_method('spawn')
    if len(sys.argv) != 2:
        print("Usage: python main.py <config_file>")
        sys.exit(1)

    config_file = sys.argv[1]
    main(config_file)