遗传算法求解最优化问题python
时间: 2023-12-06 16:38:20 浏览: 171
遗传算法求最优解
遗传算法是一种基于自然选择和遗传学原理的优化算法,可以用于求解最优化问题。在Python中,可以使用多个库来实现遗传算法,例如Geatpy、DEAP等。以下是使用Geatpy库实现遗传算法求解最优化问题的示例代码:
```python
import geatpy as ea
# 定义问题类
class MyProblem(ea.Problem):
def __init__(self):
name = 'MyProblem' # 问题名称
M = 2 # 目标维数
maxormins = [1, 1] # 目标最小化或最大化标记,1表示最小化,-1表示最大化
Dim = 2 # 决策变量维数
varTypes = [0, 0] # 决策变量类型,0表示实数,1表示整数
lb = [-5, -5] # 决策变量下界
ub = [5, 5] # 决策变量上界
lbin = [1, 1] # 决策变量是否包含下界,1表示包含,0表示不包含
ubin = [1, 1] # 决策变量是否包含上界,1表示包含,0表示不包含
self.M = M
self.maxormins = maxormins
self.Dim = Dim
self.varTypes = varTypes
self.lb = lb
self.ub = ub
self.lbin = lbin
self.ubin = ubin
self.name = name
def aimFunc(self, pop): # 目标函数
x1 = pop.Phen[:, [0]]
x2 = pop.Phen[:, [1]]
f1 = x1**2 + x2**2
f2 = (x1-2)**2 + (x2-2)**2
pop.ObjV = np.hstack([f1, f2])
# 初始化问题对象
problem = MyProblem()
# 初始化算法模板对象
algorithm = ea.soea_NSGA2_templet(problem, populationType='real', selectionType='tourSel',
crossoverType='realSBX', mutationType='realPolynomial',
maxGen=100, printLog=True)
# 运行算法
algorithm.run()
# 输出结果
best_ind = algorithm.bestInd
print('最优解:', best_ind.Phen)
print('最优目标函数值:', best_ind.ObjV)
```
上述代码中,定义了一个MyProblem类来表示问题,其中包括问题名称、目标维数、目标最小化或最大化标记、决策变量维数、决策变量类型、决策变量下界、决策变量上界等信息。在类中还定义了目标函数aimFunc,用于计算目标函数值。然后,使用NSGA-II算法模板来求解问题,最终输出最优解和最优目标函数值。
阅读全文