在Python中将keras模型作为参数发送到外部函数

我想在培训后将Keras模型发送到另一个python函数，该函数保存在另一个python文件中？如何将模型作为参数发送？谢谢。

如果我理解正确，您希望将脚本A中创建的模型转换为脚本B，以便在脚本B中使用

根据我的经验，在不同脚本中使用Keras模型的最简单方法是将模型作为文件保存到磁盘

从keras.models导入负载模型
model.save（'my_model.h5'）#创建一个HDF5文件'my_model.h5'
del model#删除现有模型
#返回已编译的模型
#与前一个相同
模型=负载模型（'my\u model.h5'）

将模型传递到另一个Python文件（即通过命令行参数）可以通过将保存该模型的文件名传递到第二个脚本来完成。然后，该脚本可以从磁盘加载模型并使用它

如果一次只有一个模型，您可以选择一个文件名并将其硬编码到函数中。例如：

脚本A

#假设您已经在“model”中存储了一个模型
model.save（'my\u storaged\u model.h5'）

脚本B（访问保存的模型）

从keras.models导入负载模型
def函数_a（）：
model=load\u model（'my\u storaged\u model.h5'）
回报模型。预测（…）