- TensorFlow知识图谱实战
- 王晓华
- 185字
- 2022-07-27 17:22:22
2.1.4 使用保存的Keras模式对模型进行复用
前面已经说过,对于保存的文件,Keras是将所有的信息都保存在h5文件中,包含所有的模型结构信息和训练过的参数信息。
new_model = tf.keras.models.load_model('./saver/the_save_model.h5')
tf.keras.models.load_model函数从给定的地址中载入h5模型,载入完成后会依据存档自动建立一个新的模型。
模型的复用可直接调用模型predict函数:
new_prediction = new_model.predict(iris_data)
这里直接将iris数据作为预测数据进行输入。全部代码如下所示。
【程序2-3】
![](https://epubservercos.yuewen.com/281CEB/23721624209516806/epubprivate/OEBPS/Images/Figure-P47_8276.jpg?sign=1739531058-sBY4SOkWxsOVZ1jUtD4N3KavwJNnY8K1-0-f12801b1b616d9066a0a406a9ee86bb1)
计算结果被完整打印出来,如图2.7所示。
![](https://epubservercos.yuewen.com/281CEB/23721624209516806/epubprivate/OEBPS/Images/Figure-P47_4421.jpg?sign=1739531058-1ySotnfNk9KOT4KXKosj4rvO8D99eRnf-0-ddfcc4e59c9bb9d977aba2c9f4c98865)
图2.7 打印结果