HeurAgenix/evolve_heuristic.py at main · idg2/HeurAgenix · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
import argparse
import os
import re
from src.pipeline.heuristic_evolver import HeuristicEvolver
from src.util.util import search_file
from src.util.llm_client.get_llm_client import get_llm_client


def parse_arguments():
    problem_pool = [problem for problem in os.listdir(os.path.join("src", "problems")) if problem != "base"]

    parser = argparse.ArgumentParser(description="Evolve heuristic")
    parser.add_argument("-p", "--problem", choices=problem_pool, required=True, help="Specifies the type of combinatorial optimization problem.")
    parser.add_argument("-m", "--smoke_test", action='store_true', help="Optional flag to conduct a preliminary smoke test.")
    parser.add_argument("-l", "--llm_config_file", type=str, default=os.path.join("data", "llm_config", "azure_gpt_4o.json"), help="Path to the language model configuration file. Default is azure_gpt_4o.json.")
    parser.add_argument("-e", "--seed_heuristic", type=str, required=True, help="The initial seed heuristic to be evolved.")
    parser.add_argument("-ed", "--evolution_dir", type=str, default="train_data", help="Directory containing the training dataset used for heuristic evolution.")
    parser.add_argument("-vd", "--validation_dir", type=str, default=None, help="Directory containing the validation dataset used to evaluate heuristic performance.")
    parser.add_argument("-pe", "--perturbation_heuristic", type=str, default=None, help="Optional name or path of an additional perturbation heuristic to explore more diverse strategies. Default is random_xxxx.py in src\problems\{problem}\heuristic\basic_heuristic")
    parser.add_argument("-pr", "--perturbation_ratio", type=float, default=0.1, help="Proportion of operations to be randomly altered during each perturbation cycle. Default is 0.1.")
    parser.add_argument("-pt", "--perturbation_time", type=float, default=1000, help="Maximum number of perturbation cycles performed per evolution round.Default is 1000.")
    parser.add_argument("-i", "--max_refinement_round", type=int, default=5, help="Total number of refinement rounds to iteratively improve heuristics. Default is 5.")
    parser.add_argument("-f", "--filter_num", type=int, default=1, help="Number of top-performing heuristics to retain after each validation phase. Default is 3.")
    parser.add_argument("-r", "--evolution_rounds", type=int, default=3, help="Total number of evolutionary iterations to perform. Default is 3.")

    return parser.parse_args()

def main():
    args = parse_arguments()
    problem = args.problem
    basic_heuristic_file = args.seed_heuristic
    evolution_dir = args.evolution_dir
    validation_dir = args.validation_dir if args.validation_dir else args.evolution_dir
    perturbation_heuristic_file = args.perturbation_heuristic
    perturbation_ratio = args.perturbation_ratio
    perturbation_time = args.perturbation_time
    max_refinement_round = args.max_refinement_round
    filter_num = args.filter_num
    evolution_rounds = args.evolution_rounds
    smoke_test = args.smoke_test
    llm_config_file = args.llm_config_file

    if not os.path.exists(evolution_dir):
        evolution_dir = search_file(evolution_dir, problem)
    if not os.path.exists(validation_dir):
        validation_dir = search_file(validation_dir, problem)
    if not basic_heuristic_file.endswith(".py"):
        basic_heuristic_file += ".py"
    basic_heuristic_file = search_file(basic_heuristic_file, problem)
    if perturbation_heuristic_file is None:
        perturbation_heuristic_file = [file_name for file_name in os.listdir(os.path.join("src", "problems", problem, "heuristics", "basic_heuristics")) if re.match( r"random_....\.py", file_name)]
        if perturbation_heuristic_file == []:
            raise Exception("No perturbation heuristics")
        perturbation_heuristic_file = os.path.join("src", "problems", problem, "heuristics", "basic_heuristics", perturbation_heuristic_file[0])
    else:
        if not perturbation_heuristic_file.endswith(".py"):
            perturbation_heuristic_file += ".py"
        perturbation_heuristic_file = search_file(perturbation_heuristic_file, problem)

    prompt_dir = os.path.join("src", "problems", "base", "prompt")
    llm_client = get_llm_client(llm_config_file, prompt_dir, None)

    heuristic_evolver = HeuristicEvolver(llm_client, problem, evolution_dir, validation_dir)
    evolved_heuristics = heuristic_evolver.evolve(
        basic_heuristic_file,
        perturbation_heuristic_file,
        perturbation_ratio=perturbation_ratio,
        perturbation_time=perturbation_time,
        max_refinement_round=max_refinement_round,
        filtered_num=filter_num,
        evolution_round=evolution_rounds,
        smoke_test=smoke_test
    )
    print(evolved_heuristics)

if __name__ == "__main__":
    main()