利用 DEAP 获取每一代最佳个体-Python教程-PHP中文网

利用 DEAP 获取每一代最佳个体

碧海醫心

发布： 2025-08-08 19:44:14

原创

584人浏览过

利用 deap 获取每一代最佳个体

本文将介绍如何使用 DEAP 库在遗传算法的每一代中高效地获取最佳个体。通过结合 HallOfFame 类和 MultiStatistics 类，可以简化代码并显著提高性能，从而方便地追踪和分析每一代的最优解。

在遗传算法的开发过程中，经常需要追踪每一代中的最佳个体，以便进行分析、可视化或用于其他目的。DEAP 库提供了一些强大的工具，可以帮助我们实现这一目标。

使用 HallOfFame 类

HallOfFame 类是 DEAP 中用于存储迄今为止找到的最佳个体的容器。它会自动跟踪种群中的最佳个体，并允许你访问它们。要使用 HallOfFame，首先需要创建一个实例，并指定要存储的最佳个体数量：

from deap import tools

hof = tools.HallOfFame(maxsize=1) # 只保留最佳个体

登录后复制

在遗传算法的循环中，HallOfFame 会自动更新。在算法执行完毕后，可以通过索引访问最佳个体：

best_individual = hof[0] # 获取最佳个体

登录后复制

结合 MultiStatistics 类

为了在每一代中获取最佳个体，我们可以结合 HallOfFame 类和 MultiStatistics 类。MultiStatistics 类允许我们注册自定义的统计函数，这些函数将在每一代中被调用。

首先，定义统计函数，该函数从 HallOfFame 中获取最佳个体：

def get_best_individual(halloffame):
    return halloffame[0] if halloffame else None

登录后复制

然后，注册 HallOfFame 和统计函数到 MultiStatistics 中：

幻舟AI

专为短片创作者打造的AI创作平台

279

查看详情

from deap import tools
import numpy as np

hof = tools.HallOfFame(maxsize=1)
stats = tools.Statistics(lambda ind: ind.fitness.values)
stats.register('min', np.min)
stats.register('mean', np.mean)
history = tools.Statistics(lambda ind: ind)
history.register('hof', get_best_individual)  # 使用hof的第一个元素
mstats = tools.MultiStatistics(fitness=stats, history=history)

登录后复制

现在，在运行遗传算法时，将 mstats 传递给 eaSimple 函数：

from deap import algorithms

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

登录后复制

在 logbook 中，你将找到每一代最佳个体的历史记录。可以这样访问它们：

best_individuals_history = logbook.chapters['history'].select('hof')

登录后复制

完整示例

以下是一个完整的示例，演示了如何使用 DEAP 获取每一代最佳个体：

import random
import numpy as np
from deap import base, creator, tools, algorithms

# 定义问题
def eval_func(individual):
    return sum(individual),

# 创建类型
creator.create("FitnessMax", base.Fitness, weights=(1.0,))
creator.create("Individual", list, fitness=creator.FitnessMax)

# 初始化
toolbox = base.Toolbox()
toolbox.register("attr_bool", random.randint, 0, 1)
toolbox.register("individual", tools.initRepeat, creator.Individual, toolbox.attr_bool, n=10)
toolbox.register("population", tools.initRepeat, list, toolbox.individual)

# 注册算子
toolbox.register("evaluate", eval_func)
toolbox.register("mate", tools.cxTwoPoint)
toolbox.register("mutate", tools.mutFlipBit, indpb=0.05)
toolbox.register("select", tools.selTournament, tournsize=3)

# 定义统计
hof = tools.HallOfFame(maxsize=1)
stats = tools.Statistics(lambda ind: ind.fitness.values)
stats.register('min', np.min)
stats.register('mean', np.mean)
history = tools.Statistics(lambda ind: ind)
history.register('hof', lambda pop: hof[0] if hof else None)  # 使用hof的第一个元素
mstats = tools.MultiStatistics(fitness=stats, history=history)

# 运行算法
population = toolbox.population(n=50)
population, logbook = algorithms.eaSimple(population, toolbox,
                                              cxpb=0.5, mutpb=0.2,
                                              ngen=10, stats=mstats,
                                              halloffame=hof, verbose=True)

# 获取每一代最佳个体
best_individuals_history = logbook.chapters['history'].select('hof')

# 打印结果
for i, individual in enumerate(best_individuals_history):
    print(f"Generation {i}: Best Individual = {individual}")

登录后复制

注意事项