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

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

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

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

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