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

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

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

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

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