这里使用了蚁群算法求解了旅行商问题,同时结合了simpy来绘图
选择下一个食物的函数为:
probability[i] = pheromone[self.now][self.not_to_foods[i]] ** pheromone_w + (1 / distance[self.now][self.not_to_foods[i]]) ** distance_w
该条路概率权重=该点信息素^信息素权重*(1/路径长)^路径权重/总和,这里后面会用random.choices就不用除了
信息素更新为:
new_pheromone[self.route[i]][self.route[i + 1]] += pheromone_Q / self.sum_lenpheromone = np.add((1 - volatilize) * pheromone, new_pheromone)
总信息素会挥发volatilize比例,然后所有蚂蚁路径每一段加上固定释放的信息量/路径总长度
话不多说,都在代码注释里了
import random
import simpy
import matplotlib.pyplot as plt
import numpy as np# 在这里我们尝试用蚁群算法求解旅行商问题,简而言之,就是寻找一条链接所有点的最短路径最后要回到初始点
# 我想绘制一下每只蚂蚁的行走过程,所以使用上次的simpy包来模拟一下,
# 这个simpy包主要是为了更好的可视化模拟训练过程,你也可以给每只蚂蚁单独线程代替,
# 或者干脆不用管过程,只用一轮一轮的上蚂蚁,在信息素部分做点处理,光看结果也可以。总之大多要比simpy快class Ant(object): # 蚂蚁类# 说一下训练过程,有很多只蚂蚁,从随机位置出发,# 每次遵从信息素的指引,选择下一个城市,直到没有可选城市,回到初始位置然后根据路径总长度在路径上散播信息素def __init__(self): # 蚂蚁构造函数# 一只蚂蚁,只需要保存走过的食物就可以,# 这里我们为了以后整信息素时不用遍历,顺便保存一下走过的长度self.route = [] # 到达食物路径,一个例子是:[2,1,3,2]self.not_to_foods = [] # 未到达的食物点self.sum_len = 0 # 走过路径总长度self.now = 0 # 当前所在位置self.next_food = 0 # 下一个所在位置self.Initial_Position(random.randint(0,foods_n-1)) # 设定初始位置# 绘图参数self.now_position = [0, 0] # 起始到终点的向量self.proportion = env.now # 走的时间,画图就是找到起始食物点,然后加上(起始和终点的向量乘以(当前时间-出发时间)/路长)就是蚂蚁位置,def Initial_Position(self, n): # 选择初始位置'self.now = n # 当前所在的食物self.next_food = n # 下一个要到的食物self.route = []self.not_to_foods = [i for i in range(foods_n)] # 未到达的食物点self.sum_len = 0self.route.append(n) # 添加到路径self.not_to_foods.remove(n) # 初始点不会在未到的食物点def Set_Now(self, next_food): # 到下一个点self.next_food = next_food # 下一个要到的食物self.route.append(next_food) # 添加到路径self.sum_len += distance[self.now][next_food] # 总路径增加if next_food in self.not_to_foods:self.not_to_foods.remove(next_food) # 移出未到的食物点self.now_position = foods[next_food] - foods[self.now] # 走的方向self.proportion = env.now # 出发时间def Next_Food(self): # 根据信息素和路径寻找下一个食物len_not_to_foods = len(self.not_to_foods) # 没选择的食物个数probability = np.zeros(len_not_to_foods) # 选择食物的概率for i in range(len_not_to_foods):# 选一个路径的概率是该点信息素^信息素权重*(1/路径长)^路径权重/总和,这里后面会用random.choices就不用除了probability[i] = pheromone[self.now][self.not_to_foods[i]] ** pheromone_w + (1 / distance[self.now][self.not_to_foods[i]]) ** distance_wnext_food = random.choices(self.not_to_foods, probability) # 根据权重选择下一个点return next_fooddef Change_Pheromone(self): # 更新信息素矩阵,先添加到一个临时变量中,信息素矩每隔一段时间后自己会挥发并将这个变量添加进去。for i in range(foods_n):new_pheromone[self.route[i]][self.route[i + 1]] += pheromone_Q / self.sum_lennew_pheromone[self.route[i + 1]][self.route[i]] = new_pheromone[self.route[i]][self.route[i + 1]]def run(self): # 蚂蚁出动global min_distance, min_routewhile True:while self.not_to_foods: # 未空还能找食物next_food = self.Next_Food()[0]self.Set_Now(next_food) # 根据信息素和路径去下一个食物位置yield env.timeout(distance[self.now][next_food]) # 等待到达终点self.now = next_food # 到达位置# 没有食物可以去了next_food = self.route[0]self.Set_Now(next_food) # 去初始点yield env.timeout(distance[self.now][next_food]) # 等待到达终点self.now = next_food # 到达位置self.Change_Pheromone() # 到了初始点之后,更新信息素if min_distance > self.sum_len:min_distance = self.sum_lenmin_route = self.routeself.Initial_Position(random.randint(0,foods_n-1)) # 重新初始化,再来一轮def Change_Pheromone(env):global pheromone, new_pheromonewhile True:# 每隔一段时间,挥发,更新信息素,# 我想要平均走完一波蚂蚁更新一波,那么,我大概需要等待平均总路长的时间(因为假设蚂蚁1m/s)这里就用(foods_n+1)/2,两点间平均距离大概是0.5yield env.timeout((foods_n + 1) / 2)pheromone = np.add((1 - volatilize) * pheromone, new_pheromone)new_pheromone = np.zeros((foods_n, foods_n))# 初始化函数,初始化蚁群,食物,信息素矩阵,
def Initialization(ants_n, foods_n, dimension):''':param ants_n: 蚁群大小:param foods_n: 食物数目:param dimension: 维度:return: 初始化蚁群,食物,信息素矩阵,距离矩阵'''ants = [Ant() for _ in range(ants_n)] # 蚁群np.random.seed(0) # 随机数种子,让生成的位置一致,你可以删去foods = np.random.rand(foods_n, dimension) # 所有食物位置,支持高维,不过绘图只有二维(python三维图太卡了),只测试二维pheromone = np.zeros((foods_n, foods_n))+2 # 信息素矩阵初始为路径长度平均数的倒数之类的distance = np.zeros((foods_n, foods_n)) # 距离矩阵for i in range(foods_n): # 计算欧氏距离for j in range(foods_n):if i > j: # 以前算过了,抄过来distance[i][j] = distance[j][i]continueif i == j: # 一样的位置就是0,不算了continuedistance[i][j] = np.sqrt(np.sum((foods[i] - foods[j]) ** 2))return ants, foods, pheromone, distancedef plt_Refresh(): # 绘图while True:plt.clf() # 清屏plt.xlim(0, 1)plt.ylim(-0.1, 1)# 绘图plt.text(0.05, -0.05, "now_time = " + str(env.now))for i in range(foods_n): # 所有的食物点for j in range(i):plt.plot([foods[i][0],foods[j][0]],[foods[i][1],foods[j][1]],linewidth=pheromone[i][j]/100)plt.scatter(foods[i][0], foods[i][1], 100)plt.text(foods[i][0], foods[i][1], i)for i in range(ants_n): # 所有的蚂蚁点# 到起始食物点,然后加上(起始和终点的向量乘以(当前时间-出发时间)/路长)就是蚂蚁位置,if ants[i].now == ants[i].next_food:position = foods[ants[i].now]else:position = foods[ants[i].now] + ants[i].now_position * (env.now - ants[i].proportion) / \distance[ants[i].now][ants[i].next_food]plt.scatter(position[0], position[1])plt.text(position[0], position[1], i)# 刷新图形plt.draw()plt.pause(time_particles)yield env.timeout(time_particles)def Firing(env): # env启动env.process(Change_Pheromone(env)) # 启动信息素矩阵env.process(plt_Refresh()) # 启动绘图for i in range(ants_n): # 启动所有蚂蚁env.process(ants[i].run())ants_n = 15 # 蚂蚁数目,一般1.5*foods_n
foods_n = 10 # 食物数目
dimension = 2 # 维度,建议2,因为绘图只搞了2
min_distance = float("inf") # 最小距离
min_route = [] # 最小路径
# 一些超参数
pheromone_w = 1 # 信息素权重
distance_w = 6 # 距离权重
volatilize = 0.5 # 信息素挥发比例
pheromone_Q = 10 # 总信息素
run_time = 100 # 模拟时间
time_particles = 0.05 # 绘图间隔
new_pheromone = np.zeros((foods_n, foods_n))
# 绘图
plt.figure()
plt.pause(10) # 方便我录屏,等10s再出画面你们要删去
env = simpy.Environment() # 设置环境并启动模拟
ants, foods, pheromone, distance = Initialization(ants_n, foods_n, dimension) # 初始化,通常,蚁群数目是食物数目1.5倍
print(foods)
Firing(env) # 开火,启动(添加要模拟的函数)
env.run(until=run_time) # 运行模拟print(min_route,min_distance)
plt.show() # 遍历完成后不消失
自己用建议删去 plt.pause(10)这一行,同时,可以调大time_particles,
这是结果:
视频审核还没通过,通过了就放出来