DEAP:使个体的健康取决于整个种群

DEAP: make fitness of an individual depend on entire population

要实施共享策略(描述为 here),适应度函数需要依赖于种群中的其他个体。

def shared_fitness(individual, population):
    #compute
    return result

如何在遗传算法使用的工具箱中注册此函数?为了做到这一点:

toolbox = base.Toolbox()
toolbox.register('evaluate', shared_fitness, population=pop)

我首先要定义一个群体pop。但是在算法时期,我希望用当前种群而不是初始种群来评估适应度。

如何实现依赖于种群中其他个体的适应度函数?

请注意,在 99% 的时间里,我们的人口将是某种列表(或其他容器对象)。当我们将这些对象传递给函数时,我们传递的是指针而不是值。这意味着我们对种群所做的任何更改都会影响评估函数,该函数包含指向种群的指针。
为了进行健全性测试,我采用了 N-Queens example from DEAP 并对评估函数做了一个小改动——只是为了打印当前的前 5 名人口成员。当您 运行 时,您可以看到输出发生变化,即使评估函数收到 "initial population" 作为输入。

如果出于某种原因您的人口是按值而不是指针传递的,那么一个始终包含当前人口的全局变量可能会有所帮助,尽管这当然不太可取。

#    This file is part of DEAP.
#
#    DEAP is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Lesser General Public License as
#    published by the Free Software Foundation, either version 3 of
#    the License, or (at your option) any later version.
#
#    DEAP is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
#    GNU Lesser General Public License for more details.
#
#    You should have received a copy of the GNU Lesser General Public
#    License along with DEAP. If not, see <http://www.gnu.org/licenses/>.

import random

import numpy

from deap import algorithms
from deap import base
from deap import creator
from deap import tools

# Problem parameter
NB_QUEENS = 20
INDIV_COUNT = 0


def evalNQueens(individual, population):
    global INDIV_COUNT
    """Evaluation function for the n-queens problem.
    The problem is to determine a configuration of n queens
    on a nxn chessboard such that no queen can be taken by
    one another. In this version, each queens is assigned
    to one column, and only one queen can be on each line.
    The evaluation function therefore only counts the number
    of conflicts along the diagonals.
    """
    size = len(individual)
    # Count the number of conflicts with other queens.
    # The conflicts can only be diagonal, count on each diagonal line
    left_diagonal = [0] * (2 * size - 1)
    right_diagonal = [0] * (2 * size - 1)

    # Sum the number of queens on each diagonal:
    for i in range(size):
        left_diagonal[i + individual[i]] += 1
        right_diagonal[size - 1 - i + individual[i]] += 1

    # Count the number of conflicts on each diagonal
    sum_ = 0
    for i in range(2 * size - 1):
        if left_diagonal[i] > 1:
            sum_ += left_diagonal[i] - 1
        if right_diagonal[i] > 1:
            sum_ += right_diagonal[i] - 1

    if INDIV_COUNT % len(population) == 0:
        print(f'top 5 individuals @ generation {int(INDIV_COUNT / 300)}: {population[:5]}')
    INDIV_COUNT += 1

    return sum_,


creator.create("FitnessMin", base.Fitness, weights=(-1.0,))
creator.create("Individual", list, fitness=creator.FitnessMin)

# Since there is only one queen per line,
# individual are represented by a permutation
toolbox = base.Toolbox()
toolbox.register("permutation", random.sample, range(NB_QUEENS), NB_QUEENS)

# Structure initializers
# An individual is a list that represents the position of each queen.
# Only the line is stored, the column is the index of the number in the list.
toolbox.register("individual", tools.initIterate, creator.Individual, toolbox.permutation)
toolbox.register("population", tools.initRepeat, list, toolbox.individual)

toolbox.register("mate", tools.cxPartialyMatched)
toolbox.register("mutate", tools.mutShuffleIndexes, indpb=2.0 / NB_QUEENS)
toolbox.register("select", tools.selTournament, tournsize=3)


def main(seed=0):
    random.seed(seed)

    pop = toolbox.population(n=300)
    toolbox.register("evaluate", evalNQueens, population=pop)
    hof = tools.HallOfFame(1)
    stats = tools.Statistics(lambda ind: ind.fitness.values)
    stats.register("Avg", numpy.mean)
    stats.register("Std", numpy.std)
    stats.register("Min", numpy.min)
    stats.register("Max", numpy.max)

    algorithms.eaSimple(pop, toolbox, cxpb=0.5, mutpb=0.2, ngen=100, stats=stats,
                        halloffame=hof, verbose=True)

    return pop, stats, hof


if __name__ == "__main__":
    main()