pymoo_nsga2_zdt1_multiobjective_optimization_pareto_front_visualization.py

1from pymoo.algorithms.moo.nsga2 import NSGA2
2from pymoo.problems import get_problem
3from pymoo.optimize import minimize
4from pymoo.visualization.scatter import Scatter
5
6# 1. Prepare the problem
7problem = get_problem("zdt1")
8
9# 2. Configure the algorithm
10algorithm = NSGA2(pop_size=100)
11
12# 3. Run the optimization
13res = minimize(problem,
14               algorithm,
15               ('n_gen', 200),
16               seed=1,
17               verbose=False)
18
19# 4. Visualize the results
20plot = Scatter()
21plot.add(problem.pareto_front(), plot_type="line", color="black", alpha=0.7)
22plot.add(res.F, facecolor="none", edgecolor="red")
23plot.show()