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
|
#!/usr/bin/env python3
import math
import time
import random
Nodes = [
[0, 10, 20, 30, 40],
[10, 0, 20, 30, 40],
[20, 20, 0, 30, 40],
[30, 30, 30, 0, 40],
[40, 40, 40, 40, 0],
]
Pheromone = [
[0, 0, 0, 0, 0],
[0, 0, 0, 0, 0],
[0, 0, 0, 0, 0],
[0, 0, 0, 0, 0],
[0, 0, 0, 0, 0],
]
class Ant:
def __init__(self, node):
self.node = node
self.route = int()
self.exclusion_list = list()
def __repr__(self):
return "{node: %s}" % (self.node)
def __str__(self):
return "{node: %s}" % (self.node)
def nextHop(self):
n = math.ceil(random.random() * len(Nodes)) - 1
if (
n not in self.exclusion_list
and n != self.exclusion_list[-1]
and Nodes[self.exclusion_list[-1]][n] != 0
):
self.exclusion_list.append(n)
return n
else:
if len(self.exclusion_list) == len(Nodes):
return None
else:
self.nextHop()
def nextHop2(self):
pass
def walk(self):
self.exclusion_list.append(self.node)
for _ in range(0, len(Nodes)):
next = self.nextHop()
if next is not None:
self.route += Nodes[self.node][next]
self.node = next
def printRoute(self):
print(self.exclusion_list)
class ACO:
def __init__(self, ant_count):
self.ant_count = ant_count
self.Ants = self.factory()
def factory(self):
result = list()
for _ in range(0, self.ant_count):
node = math.ceil(random.random() * 5) - 1
result.append(Ant(node))
return result
def fitness(self):
pass
def run(self):
for ant in self.Ants:
ant.walk()
ant.printRoute()
for i in range(0, len(Nodes) - 1):
Pheromone[ant.exclusion_list[i]][ant.exclusion_list[i + 1]] = (
1 / ant.route
)
def main():
random.seed(time.time())
rho = 0
alpha = 1
beta = 1
aco = ACO(5)
# for ant in aco.Ants:
# print("ant:", ant)
aco.run()
for i in range(0, len(Pheromone)):
for j in range(0, len(Pheromone)):
print(Pheromone[i][j], end=" ")
print()
if __name__ == "__main__":
main()
|
