使用 Keras 進行深度學習 - 儲存模型

我們將當前工作目錄中的 models 資料夾中我們的本地驅動器中儲存已訓練的模型。要儲存模型，請執行以下程式碼 -

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

執行程式碼後的輸出如下所示 -

現在，由於您已儲存了經過訓練的模型，因此以後可以使用該模型來處理未知資料。