如何将keras模型与其他数据一起保存并一起加载？

Question

我有以下代码来训练keras神经网络

docker-compose up -d
Creating network "test_frontend" with the default driver
Creating network "test_backend" with the default driver
Creating test_mysql_1 ... done
Creating test_phpmyadmin_1 ... done

如您所见，它会将模型保存到h5文件中，但是仅保存keras模型。 如何同时保存其他数据（如指标），然后在加载keras模型时也能够加载它们。

谢谢！

Answer 1

您可以使用任何序列化框架来执行此操作。

import hickle

def save(self, path):
      self.model.save(path)
      hkl.dump(self.metrics, 'metrics.hkl', mode='w')

def load(self, path):
      self.model = load_model(path)
      self.metrics = hkl.load('metrics.hkl')

您还可以将其保存为单个文件，只需从度量标准和模型对象中列出一个列表或另一个对象即可。我建议分开保存。