-
Notifications
You must be signed in to change notification settings - Fork 7
/
slowRVEAtests.py
155 lines (145 loc) · 5.29 KB
/
slowRVEAtests.py
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
from pyrvea.Population.Population import Population
from pyrvea.Problem.testproblem import TestProblem
from pyrvea.EAs.slowRVEA import slowRVEA
import altair as alt
import numpy as np
def main():
problems = ["DTLZ2", "DTLZ3"]
numobjs = [3, 6, 8, 10]
algorithms = {"SmoothRVEA": smoothEvolve, "AbruptRVEA": abruptEvolve}
foldername = "./results/"
for problem_name in problems:
for numobj in numobjs:
name = problem_name
k = 10
numconst = 0
numvar = numobj + k - 1
problem = TestProblem(name, numvar, numobj, numconst)
orig_point = [1] * numobj
first_ref = [1, 1] + [0] * (numobj - 2)
second_ref = [0] * (numobj - 2) + [1, 1]
for algo_name, evolve in algorithms.items():
filename = foldername + name + "_" + str(numobj) + "_" + algo_name + "_"
archive_df = evolve(problem, orig_point, first_ref, second_ref)
objective_norms = archive_df["objective_values"].apply(
lambda x: np.linalg.norm(x)
)
angle_dev_from_1 = archive_df["objective_values"].apply(
lambda x: np.degrees(
np.arccos(
np.dot(x, first_ref)
/ (np.linalg.norm(x) * np.linalg.norm(first_ref))
)
)
)
angle_dev_from_2 = archive_df["objective_values"].apply(
lambda x: np.degrees(
np.arccos(
np.dot(x, second_ref)
/ (np.linalg.norm(x) * np.linalg.norm(second_ref))
)
)
)
archive_df["objective_norms"] = objective_norms
archive_df["angle_1"] = angle_dev_from_1
archive_df["angle_2"] = angle_dev_from_2
x = alt.X("generation")
y_obj = alt.Y(
"median(objective_norms)",
scale=alt.Scale(type="log", domain=(1, 10)),
)
y_angle_1 = alt.Y("median(angle_1)")
y_angle_2 = alt.Y("median(angle_2)")
magnitude = (
alt.Chart(archive_df)
.mark_line(clip=True)
.encode(x=x, y=y_obj)
.properties(title="Median Magnitude of objective vectors")
)
angle_1 = (
alt.Chart(archive_df)
.mark_line(clip=True)
.encode(x=x, y=y_angle_1)
.properties(
title="Median Angular Deviation of objective vectors from \
Reference Point 1"
)
)
angle_2 = (
alt.Chart(archive_df)
.mark_line(clip=True)
.encode(x=x, y=y_angle_2)
.properties(
title="Median Angular Deviation of objective vectors from \
Reference Point 2"
)
)
magnitude.save(filename + "magnitude.html")
angle_1.save(filename + "angle_1.html")
angle_2.save(filename + "angle_2.html")
def smoothEvolve(problem, orig_point, first_ref, second_ref):
"""Evolves using RVEA with abrupt change of reference vectors."""
pop = Population(problem, assign_type="empty", plotting=False)
try:
pop.evolve(slowRVEA, {"generations_per_iteration": 200, "iterations": 15})
except IndexError:
return pop.archive
try:
pop.evolve(
slowRVEA,
{
"generations_per_iteration": 10,
"iterations": 20,
"old_point": orig_point,
"ref_point": first_ref,
},
)
except IndexError:
return pop.archive
try:
pop.evolve(
slowRVEA,
{
"generations_per_iteration": 10,
"iterations": 20,
"old_point": first_ref,
"ref_point": second_ref,
},
)
except IndexError:
return pop.archive
return pop.archive
def abruptEvolve(problem, orig_point, first_ref, second_ref):
"""Evolves using RVEA with abrupt change of reference vectors."""
pop = Population(problem, assign_type="empty", plotting=False)
try:
pop.evolve(slowRVEA, {"generations_per_iteration": 200, "iterations": 15})
except IndexError:
return pop.archive
try:
pop.evolve(
slowRVEA,
{
"generations_per_iteration": 10,
"iterations": 20,
"old_point": first_ref,
"ref_point": first_ref,
},
)
except IndexError:
return pop.archive
try:
pop.evolve(
slowRVEA,
{
"generations_per_iteration": 10,
"iterations": 20,
"old_point": second_ref,
"ref_point": second_ref,
},
)
except IndexError:
return pop.archive
return pop.archive
if __name__ == "__main__":
main()