我正在尝试从DEAP 模块eaSimple
运行算法。我想指定每一代要选择的个体数量。但是,如果我为选择函数指定一个参数,则会出现错误。k
from deap import base, tools, creator, algorithms
import random
import numpy as np
def fitness(individual):
x = individual[0]
y = individual[1]
return np.exp(-9*x*y) * np.sin(3*np.pi*x)**2 * np.sin(3*np.pi*y)**2,
toolbox = base.Toolbox()
creator.create("FitnessMax", base.Fitness, weights=(1.0,))
creator.create("Individual", list, fitness=creator.FitnessMax)
toolbox.register("individual", tools.initRepeat, creator.Individual, random.random, n=2)
toolbox.register("population", tools.initRepeat, list, toolbox.individual)
toolbox.register('evaluate', fitness)
toolbox.register('mutate', tools.mutPolynomialBounded, eta=.6, low=[0,0], up=[1,1], indpb=0.1)
toolbox.register('mate', tools.cxUniform, indpb=0.5)
toolbox.register('select', tools.selBest, k=50)
pop = toolbox.population(n=100)
pop, logbook = algorithms.eaSimple(pop, toolbox, cxpb=0.5, mutpb=0.1, ngen=100)
此示例中的最后一行引发了错误
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-2-5ee0faee2c49> in <module>
----> 1 pop, logbook = algorithms.eaSimple(pop, toolbox, cxpb=0.5, mutpb=0.1, ngen=100)
/usr/local/lib64/python3.7/site-packages/deap/algorithms.py in eaSimple(population, toolbox, cxpb, mutpb, ngen, stats, halloffame, verbose)
163 for gen in range(1, ngen + 1):
164 # Select the next generation individuals
--> 165 offspring = toolbox.select(population, len(population))
166
167 # Vary the pool of individuals
TypeError: selBest() got multiple values for argument 'k'
请注意,将行替换为toolbox.register('select', tools.selBest, k=50)
删除toolbox.register('select', tools.selBest)
错误。
这里的默认值是什么k
,我如何指定我自己的值k
?