flowshop_transformer / source /launch_process_dataset.py
younadi's picture
commit
6179bb6
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)