一、基本步驟
遺傳算法求解旅行商問題(TSP)的一般步驟如下:
-
編碼:
- 通常采用整數編碼,將城市的訪問順序表示為一個染色體。例如,假設有 5 個城市,編碼為[1, 3, 5, 2, 4],表示旅行商的訪問順序為城市 1、城市 3、城市 5、城市 2、城市 4,最后回到出發城市 1。
-
初始化種群:
- 隨機生成一定數量的染色體,作為初始種群。
-
適應度評估:
- 計算每個染色體所代表的路徑長度,路徑長度越短,適應度越高。適應度函數可以是路徑長度的倒數或其他與路徑長度相關的函數。
-
選擇操作:
- 基于適應度,選擇一定比例的染色體進入下一代種群。常見的選擇方法有輪盤賭選擇、錦標賽選擇等。
-
交叉操作:
- 對選中的染色體進行交叉,生成新的染色體。例如,單點交叉,隨機選擇一個交叉點,交換兩個染色體在交叉點之后的部分。
-
變異操作:
- 以一定的概率對染色體中的基因進行變異,例如隨機交換兩個城市的位置。
-
重復步驟 3 - 6 ,直到滿足終止條件(如達到預定的迭代次數或找到滿意的解)。
通過不斷迭代,種群中的染色體逐漸優化,最終得到較優的 TSP 路徑。需要注意的是,遺傳算法的參數(如種群大小、交叉概率、變異概率等)需要根據具體問題進行調整和優化,以獲得更好的求解效果。
二、代碼
#!/usr/bin/env python
# coding: utf-8# In[32]:import numpy as np
import copy
import matplotlib
import matplotlib.pyplot as plt
from matplotlib.pyplot import MultipleLocator
import random# In[33]:#準備好距離矩陣
city_num = 5
city_dist_mat = np.zeros([city_num, city_num])
city_dist_mat[0][1] = city_dist_mat[1][0] = 1165
city_dist_mat[0][2] = city_dist_mat[2][0] = 1462
city_dist_mat[0][3] = city_dist_mat[3][0] = 3179
city_dist_mat[0][4] = city_dist_mat[4][0] = 1967
city_dist_mat[1][2] = city_dist_mat[2][1] = 1511
city_dist_mat[1][3] = city_dist_mat[3][1] = 1942
city_dist_mat[1][4] = city_dist_mat[4][1] = 2129
city_dist_mat[2][3] = city_dist_mat[3][2] = 2677
city_dist_mat[2][4] = city_dist_mat[4][2] = 1181
city_dist_mat[3][4] = city_dist_mat[4][3] = 2216
#標號說明
#list_city = ['0_北京', '1_西安', '2_上海', '3_昆明', '4_廣州']# In[34]:#1.定義個體類,包括基因(城市路線)和適應度
num_person_idx = 0
num_person = 0
dis_list = []
class Individual:def __init__(self, genes = None):global num_personglobal dis_listglobal num_person_idxnum_person_idx += 1if num_person_idx % 20 == 0:num_person += 1self.genes = genesif self.genes == None:genes = [0]*5temp = [0]*4temp = [i for i in range(1,city_num)]########################################################################random.shuffle(temp)genes[1:] = tempgenes[0] = 0self.genes = genesself.fitness = self.evaluate_fitness()else:self.fitness = float(self.evaluate_fitness())#2. #計算個體的適應度def evaluate_fitness(self):dis = 0for i in range(city_num - 1):dis += city_dist_mat[self.genes[i]][self.genes[i+1]]if i == city_num - 2:dis += city_dist_mat[self.genes[i + 1]][0]#回到0if num_person_idx % 20 == 0:dis_list.append(dis)return 1/dis# In[35]:def copy_list(old):new = []for element in old:new.append(element)return new
def sort_win_num(group):for i in range(len(group)):for j in range(len(group) - i - 1):
# print('group[j].fitness_type', type(group[j].fitness))if group[j].fitness < group[j+1].fitness:temp = group[j]group[j] = group[j+1]group[j+1] = tempreturn group
#定義Ga類
#3~5,交叉、變異、更新種群,全部在Ga類中實現
class Ga:#input_為城市間的距離矩陣def __init__(self, input_):#聲明一個全局變量global city_dist_matcity_dist_mat = input_#當代的最佳個體########################################此處做了更改#self.best = Noneself.best = Individual(None)
# print("BBBBBBbest.fitness",self.best.fitness)#種群self.individual_list = []#每一代的最佳個體self.result_list = []#每一代個體對應的最佳適應度self.fitness_list = []#交叉,這里采用交叉變異def cross(self):new_gen = []#隨機選取一段,含有num_cross個數字(城市)num_cross = 3#后期可能需要調試的參數,考慮到實際問題里只有5個城市,所以認為3較為合適for i in range(0, len(self.individual_list) - 1, 2):parent_gen1 = copy_list(self.individual_list[i].genes)parent_gen2 = copy_list(self.individual_list[i+1].genes)
# print("parent_gen1",parent_gen1)
# print("parent_gen2",parent_gen2) index1_1 = 0index1_2 = 0index2_1 = 0index2_2 = 0#定義一個下表列表index_list = [0]*3for i in range(city_num - 3):#就是2,即0,1index_list[i] = i + 1index1_1 = random.choice(index_list)index1_2 = index1_1 + 2index2_1 = random.choice(index_list)index2_2 = index2_1 + 2choice_list1 = parent_gen1[index1_1:index1_2 + 1]choice_list2 = parent_gen2[index2_1:index2_2 + 1]
# print("choice_list1",choice_list1)
# print("choice_list2",choice_list2)#利用這一段生成兩個子代,下面的賦值只是為了獲取長度,所以用哪個父代能可以#也可以直接用city_num直接代替son_gen1 = [0]*city_numson_gen2 = [0]*city_num
# print('son_gen1_size = ',len(son_gen1))
# print('son_gen2_size = ',len(son_gen2))
# print("index1_1 == ",index1_1)
# print("index1_2 == ",index1_2)
# print("index2_1 == ",index2_1)
# print("index2_2 == ",index2_2)#找到之后進行交叉,分別得到son_gen1,son_gen2#先把選中的段復制進去son_gen1[index1_1: index1_2 + 1] = choice_list1son_gen2[index2_1: index2_2 + 1] = choice_list2
# print("now, son_gen1 = ", son_gen1)
# print("now, son_gen2 = ", son_gen2)#然后左、右“查漏補缺”temp1 = choice_list1temp2 = choice_list2if index1_1 == 0:passelse: for i in range(index1_1):for j in range(city_num):#如果父代2里面的這個當初沒被選中,那就加入son_gene1if parent_gen2[j] not in choice_list1:son_gen1[i] = parent_gen2[j]#這個時候要擴增choice_list1, 這樣parent_gen2里面未被選中的元素才會一個個被遍歷到#1choice_list1.append(parent_gen2[j])#找到之后馬上break,防止被覆蓋breakchoice_list1 = temp1if index1_2 == city_num - 1:passelse:for i in range(index1_2 + 1, city_num):for j in range(city_num):if parent_gen2[j] not in choice_list1:son_gen1[i] = parent_gen2[j]#這個時候要擴增choice_list1, 這樣parent_gen2里面未被選中的元素才會一個個被遍歷到#2choice_list1.append(parent_gen2[j])#找到之后馬上break,防止被覆蓋break#son_gen2亦是如此if index2_1 == 0:passelse:for i in range(index2_1):for j in range(city_num):#如果父代1里面的這個當初沒被選中,那就加入son_gen2if parent_gen1[j] not in choice_list2:son_gen2[i] = parent_gen1[j]#這個時候要擴增choice_list2, 這樣parent_gen1里面未被選中的元素才會一個個被遍歷到#3choice_list2.append(parent_gen1[j])#找到之后馬上break,防止被覆蓋breakchoice_list2 = temp2if index2_2 == city_num - 1:passelse:for i in range(index2_2 + 1, city_num):for j in range(city_num):if parent_gen1[j] not in choice_list2:
# print("i == ", i)son_gen2[i] = parent_gen1[j]#這個時候要擴增choice_list2, 這樣parent_gen1里面未被選中的元素才會一個個被遍歷到#4choice_list2.append(parent_gen1[j])#找到之后馬上break,防止被覆蓋break#新生成的子代基因加入new_gene列表
# print('son_gen1 = ',son_gen1)
# print('son_gen2 = ',son_gen2)new_gen.append(Individual(son_gen1))#print('new_gen[-1].genes', new_gen[-1].genes)new_gen.append(Individual(son_gen2))return new_gen#變異def mutate(self, new_gen):mutate_p = 0.02#待調參數index_list = [0]*(city_num-1)index_1 = 1index_2 = 1for i in range(city_num - 1):index_list[i] = i + 1for individual in new_gen:if random.random() < mutate_p:
# change += 1#如果變異,采用基于位置的變異,方便起見,直接使用上面定義的index列表index_l = random.choice(index_list)
# index_2 = (index_1 + 2) % city_num#這里讓間隔為2的兩個城市進行交換index_2 = random.choice(index_list)while index_1 == index_2:index_2 = random.choice(index_list)#交換temp = individual.genes[index_1]individual.genes[index_1] = individual.genes[index_2]individual.genes[index_2] = temp#變異結束,與老一代的進行合并self.individual_list += new_gen#選擇def select(self):#在此選用輪盤賭算法#考慮到5的階乘是120,所以可供選擇的個體基數應該適當大一些,#在此每次從種群中選擇6個,進行輪盤賭,初始化60個個體,同時適當調高變異的概率select_num = 6select_list = []for i in range(select_num):gambler = random.choice(self.individual_list)gambler = Individual(gambler.genes)select_list.append(gambler)#求出這些fitness之和sum = 0for i in range(select_num):sum += select_list[i].fitnesssum_m = [0]*select_num#實現概率累加for i in range(select_num):for j in range(i+1):sum_m[i] += select_list[j].fitnesssum_m[i] /= sumnew_select_list = []p_num = 0#隨機數for i in range(select_num):p_num = random.uniform(0,1)if p_num>0 and p_num < sum_m[0]:new_select_list.append(select_list[0])elif p_num>= sum_m[0] and p_num < sum_m[1]:new_select_list.append(select_list[1])elif p_num >= sum_m[1] and p_num < sum_m[2]:new_select_list.append(select_list[2])elif p_num >= sum_m[2] and p_num < sum_m[3]:new_select_list.append(select_list[3])elif p_num >= sum_m[3] and p_num < sum_m[4]:new_select_list.append(select_list[4])elif p_num >= sum_m[4] and p_num < sum_m[5]:new_select_list.append(select_list[5])else:pass#將新生成的一代替代父代種群self.individual_list = new_select_list#更新種群def next_gen(self):#交叉new_gene = self.cross()#變異self.mutate(new_gene)#選擇self.select()#獲得這一代的最佳個體
# print("**************************************")
# print('self.best.fitness = ', self.best.fitness)
# print('now, best.fitness = ', self.best.fitness)for individual in self.individual_list:if individual.fitness > self.best.fitness:self.best = individual
# print("更換了最優路徑")
# print('now, best.fitness = ', self.best.fitness)def train(self):#隨機出初代種群#individual_num = 60self.individual_list = [Individual() for _ in range(individual_num)]#迭代gen_num = 100for i in range(gen_num):#從當代種群中交叉、變異、選擇出適應度最佳的個體,獲得子代產生新的種群self.next_gen()#連接首位
# print("i = ", i)result = copy.deepcopy(self.best.genes)result.append(result[0])self.result_list.append(result)self.fitness_list.append(self.best.fitness)print(self.result_list[-1])print('距離總和是:', 1/self.fitness_list[-1])
# return self.result_list, self.fitness_listdef draw(self):x_list = [i for i in range(num_person)]y_list = dis_listplt.rcParams['figure.figsize'] = (60, 45)plt.plot(x_list, y_list, color = 'g')plt.xlabel('Cycles',size = 50)plt.ylabel('Route',size = 50)x = np.arange(0, 910, 20)y = np.arange(7800, 12000, 100)plt.xticks(x)plt.yticks(y)plt.title('Trends in distance changes', size = 50)plt.tick_params(labelsize=30)
# plt.savefig("D:\AI_pictures\遺傳算法求解TSP問題_1_輪盤賭算法")plt.show()
route = Ga(city_dist_mat)
route.train()
route.draw()