keras神经网络模型的保存与加载
一旦你利用Keras完成了訓練,你可以將你的網絡保存在HDF5里面。
keras的模型保存分為多種情況。
一、不保存模型只顯示大概結構
model.summary()
這個函數會打印模型結構,但是僅僅是打印到控制臺。
keras.utils.plot_model()
使用graphviz中的dot.exe生成網絡結構拓撲圖
二、保存模型結構
keras.models.Model
對象的to_json,to_yaml只保存模型結構,加載時使用keras.models.model_from_json(),keras.models.model_from_yaml()
keras.model.get_config()返回文本形式的配置
使用keras.model.model_from_config可以加載模型。
三、保存全部結構(最常用的方法)
keras.core.saving.py這個文件十分重要,keras的模型保存、加載都需要這個文件。
但是不建議直接使用這個文件,因為keras中的Model對象和models模塊會調用這個文件。
keras.core包下的內容一般供內部使用,不暴露給使用者。
Model對象提供了save()和save_wights()兩個方法。
只需要導入from keras.models import model_from_json就是這么簡單
基于Iris數據集如何保存model
''' @author: 毛利 ''' from sklearn import datasets import numpy as np from keras.models import Sequential from keras總結
以上是生活随笔為你收集整理的keras神经网络模型的保存与加载的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: keras 二分类
- 下一篇: Map和hashmap