如何导出python中的模型参数?这篇文章教给你两个方法( 二 )

每调用一次保存操作会创建后3个数据文件并创建一个检查点(checkpoint)文件,简单理解就是权重等参数被保存到 .chkp.data 文件中,以字典的形式;图和元数据被保存到 .chkp.meta 文件中,可以被 tf.train.import_meta_graph 加载到当前默认的图 。
(二)通过keras保存和读取模型
model.save(filepath),同时保存model和权重的import numpy as npfrom keras.datasets import mnistfrom keras.utils import np_utilsfrom keras.models import Sequentialfrom keras.layers import Densefrom keras.optimizers import SGD# 载入数据(x_train,y_train),(x_test,y_test) = mnist.load_data()# (60000,28,28)print('x_shape:',x_train.shape)# (60000)print('y_shape:',y_train.shape)# (60000,28,28)->(60000,784)x_train = x_train.reshape(x_train.shape[0],-1)/255.0x_test = x_test.reshape(x_test.shape[0],-1)/255.0# 换one hot格式y_train = np_utils.to_categorical(y_train,num_classes=10)y_test = np_utils.to_categorical(y_test,num_classes=10)# 创建模型,输入784个神经元,输出10个神经元model = Sequential([        Dense(units=10,input_dim=784,bias_initializer='one',activation='softmax')    ])# 定义优化器sgd = SGD(lr=0.2)# 定义优化器,loss function,训练过程中计算准确率model.compile(    optimizer = sgd,    loss = 'mse',    metrics=['accuracy'],)# 训练模型model.fit(x_train,y_train,batch_size=64,epochs=5)# 评估模型loss,accuracy = model.evaluate(x_test,y_test)print('test loss',loss)print('accuracy',accuracy)# 保存模型model.save('model.h5')【如何导出python中的模型参数?这篇文章教给你两个方法】以上就是小编给大家带来的导出python中的模型参数的方法,希望大家通过阅读小编的文章之后能够有所收获!如果大家觉得小编的文章不错的话,可以多多分享给有需要的人 。

推荐阅读