使用 Keras 进行深度学习 - 保存模型

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

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

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

现在，由于您已经保存了经过训练的模型，因此您稍后可以使用它来处理未知数据。