时间戳

首页 > 解决方案 > 如何在 PyGad 中将 ga_instance 加载为“.pkl”格式？

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

问题描述

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

保存功能：

文件名 = 'GNN_CPTNet' #GNN_CPTNet.pkl

ga_instance.save（文件名=文件名）

加载函数：

加载的ga_instance = pygad.load（文件名=文件名）

加载_ga_instance.plot_result()

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

标签： pythonpytorchgenetic-algorithmgenetic

解决方案

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

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

def fitness_func(solution, solution_idx):
    ...


def callback_generation(ga_instance):
    ...

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

无论如何，最好发布您用来给出更准确答案的示例代码。

感谢您使用PyGAD :)

推荐阅读