📌  相关文章
📜  使用Keras进行深度学习-保存模型

📅  最后修改于: 2020-12-11 05:06:45             🧑  作者: Mango

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

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

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

节省模型

现在，当您保存训练有素的模型时，以后可以使用它来处理未知数据。