如何在 PyGad 中将 ga_instance 加载为“.pkl”格式？

问题描述

我一直在尝试将 PyGad 训练的实例加载到另一个文件中，以便进行一些预测。但是我在加载过程中遇到了一些问题。 在训练阶段之后，我像这样保存了实例：

保存功能：

文件名 = 'GNN_CPTNet' #GNN_CPTNet.pkl

ga_instance.save(文件名=文件名)

加载函数：

loaded_ga_instance = pygad.load(filename=filename)

loaded_ga_instance.plot_result()

但是，当我尝试在新笔记本或脚本中加载实例时，我无法加载实例，尤其是“GNN_CPT Net.pkl”文件。

解决方法

在新脚本中，您应该定义原始脚本中使用的适应度函数和所有回调函数。

例如，如果您只使用了 on_generation (callback_generation) 参数，则应定义以下函数：

def fitness_func(solution,solution_idx):
    ...


def callback_generation(ga_instance):
    ...

这样，保存的实例就会被正确加载。

无论如何，最好发布您使用的示例代码以提供更准确的答案。

感谢您使用 PyGAD :)