30 lines
853 B
Python
30 lines
853 B
Python
from pymoo.algorithms.moo.nsga2 import NSGA2
|
|
from pymoo.algorithms.moo.ctaea import CTAEA
|
|
from pymoo.algorithms.moo.nsga3 import NSGA3
|
|
|
|
from pymoo.factory import get_reference_directions
|
|
from pymoo.optimize import minimize
|
|
|
|
from pymoo.visualization.scatter import Scatter
|
|
from reproblem import *
|
|
|
|
problem = CRE22()
|
|
ref_dirs = get_reference_directions("das-dennis", 2, n_partitions=64)
|
|
|
|
algorithm = NSGA2(pop_size=100)
|
|
algorithm = NSGA3(pop_size=92,
|
|
ref_dirs=ref_dirs)
|
|
# IBEA
|
|
algorithm = CTAEA(ref_dirs=ref_dirs)
|
|
|
|
res = minimize(problem,
|
|
algorithm,
|
|
('n_gen', 200),
|
|
seed=1,
|
|
verbose=False)
|
|
|
|
plot = Scatter()
|
|
plot.add(problem.pareto_front(), plot_type="line", color="black", alpha=0.7)
|
|
plot.add(res.F, facecolor="none", edgecolor="red")
|
|
plot.show()
|