一旦你利用Keras完成了訓(xùn)練，你可以將你的網(wǎng)絡(luò)保存在HDF5里面。

keras的模型保存分為多種情況。

一、不保存模型只顯示大概結(jié)構(gòu)
model.summary()
這個(gè)函數(shù)會(huì)打印模型結(jié)構(gòu)，但是僅僅是打印到控制臺(tái)。

keras.utils.plot_model()
使用graphviz中的dot.exe生成網(wǎng)絡(luò)結(jié)構(gòu)拓?fù)鋱D

二、保存模型結(jié)構(gòu)
keras.models.Model

對(duì)象的to_json,to_yaml只保存模型結(jié)構(gòu)，加載時(shí)使用keras.models.model_from_json(),keras.models.model_from_yaml()

keras.model.get_config()返回文本形式的配置
使用keras.model.model_from_config可以加載模型。

三、保存全部結(jié)構(gòu)（最常用的方法）
keras.core.saving.py這個(gè)文件十分重要，keras的模型保存、加載都需要這個(gè)文件。
但是不建議直接使用這個(gè)文件，因?yàn)閗eras中的Model對(duì)象和models模塊會(huì)調(diào)用這個(gè)文件。
keras.core包下的內(nèi)容一般供內(nèi)部使用，不暴露給使用者。

Model對(duì)象提供了save()和save_wights()兩個(gè)方法。

只需要導(dǎo)入from keras.models import model_from_json就是這么簡(jiǎn)單

基于Iris數(shù)據(jù)集如何保存model

''' @author: 毛利 ''' from sklearn import datasets import numpy as np from keras.models import Sequential from keras

總結(jié)

以上是生活随笔為你收集整理的keras神经网络模型的保存与加载的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。