我正在从头开始构建一个深度学习模型,即仅使用像numpy和matplotlib这样的核心库。我不打算在我的项目中使用任何sklearn包。
有没有办法使用我上面提到的核心Python和核心库来保存我的训练模型?
- 如果可以,我该如何保存和检索模型?
回答:
Numpy包含了几种将数组保存到文件的方法。我喜欢np.savez_compressed(),因为它允许你以压缩格式将多个数组保存到一个文件中。只需将你的训练权重和所有超参数保存到这样的.npz文件中,然后从中重建你的模型。你可以用np.load()
来检索它们。