Question

我想写出一个包含描述的keras模型（可以是一个简单的字符串），这样可以更容易地跟踪文件包含的内容。

有没有简单的方法可以做到这一点？我可以使用

保存并加载模型

model.save(filename)
[...]
model = keras.load_model(filename)

我想做什么，是

的效果

model.save(filename, description=string)
[...]
model, description = keras.load_model(filename)

关于实现这一目标的最简单方法的任何想法？

Answer 1

目前，使用Keras中当前版本的model.save无法做到这一点。它只保存模型（）的配置，例如loss function和metrics和load_model只返回已保存的内容。

此问题的一个令人讨厌的解决方案是手动将description变量添加到Keras中的类models.py，然后将所需的代码行添加到save_model和{{1} } 功能。再次，这只是一个可怕的解决方案！