欢迎访问宙启技术站
智能推送

TensorFlow.keras中load_model()函数的实例演示

发布时间:2024-01-03 00:52:47

在TensorFlow.keras中,load_model()函数是用于加载保存的模型的方法。该方法接受一个模型文件的路径作为参数,并返回加载的模型对象。

下面是一个使用load_model()函数加载模型的实例演示:

假设我们使用Keras建立了一个简单的神经网络模型,并保存为文件"model.h5":

from tensorflow import keras

# 建立模型
model = keras.Sequential([
    keras.layers.Dense(32, activation='relu', input_shape=(784,)),
    keras.layers.Dense(10, activation='softmax')
])

# 编译模型
model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

# 训练模型
model.fit(x_train, y_train, epochs=5)

# 保存模型
model.save('model.h5')

然后,我们可以使用load_model()函数加载保存的模型:

from tensorflow.keras.models import load_model

# 加载模型
loaded_model = load_model('model.h5')

# 使用加载的模型进行预测
predictions = loaded_model.predict(x_test)

在上述代码中,我们首先使用load_model()函数加载保存的模型文件"model.h5"。加载完毕后,我们可以使用加载的模型进行预测。

需要注意的是,使用load_model()函数加载模型时,需要保证模型文件的路径是正确的。如果路径错误,将会报错。

另外,load_model()函数会根据模型文件的后缀名自动选择对应的加载器。目前支持的模型文件格式有.h5、.hdf5、.json和.yml。

总结:在TensorFlow.keras中,load_model()函数能够加载保存的模型文件,并返回加载的模型对象,方便我们进行后续的预测或其他操作。