from process_dataset import process_dataset

params = [
    {
        "input_dir": f"../datasets/exhaustive_{nb_jobs}_{nb_machines}",
        "output_dir": f"../datasets/exhaustive_{nb_jobs}_{nb_machines}/top_{k_eliminated}",
        "train_ratio": 0.85,
        "seed": 97,
        "eliminate_top_k_makespans": k_eliminated,
        "duplication_factor": 0.0,
    }
    for nb_jobs in range(7, 11)
    for nb_machines in range(2, 7)
    for k_eliminated in [0, 1, 2, 3, 4]
]

for param in params:
    process_dataset(**param)