Deep Learning With Keras 简明教程

Deep Learning with Keras - Saving Model

我们将把训练好的模型保存在当前工作目录的 models 文件夹中的本地驱动器中。要保存模型，请运行以下代码：

directory = "./models/"
name = 'handwrittendigitrecognition.h5'
path = os.path.join(save_dir, name)
model.save(path)
print('Saved trained model at %s ' % path)

运行代码后的输出如下所示：

saving model

现在，由于您已经保存了训练好的模型，以后可以将其用于处理未知数据。