如何在PyGad中加载".pkl“格式的ga_instance？

Question

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

保存函数：

filename = 'GNN_CPTNet‘#GNN_CPTNet.pkl

ga_instance.save(filename=filename)

加载函数：

loaded_ga_instance = pygad.load(filename=filename)

loaded_ga_instance.plot_result()

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

Answer 1

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

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

def fitness_func(solution, solution_idx):
    ...


def callback_generation(ga_instance):
    ...

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

无论如何，最好把你用来给出更准确答案的示例代码贴出来。

感谢您使用PyGAD :)