![TensorFlow知识图谱实战](https://wfqqreader-1252317822.image.myqcloud.com/cover/115/44510115/b_44510115.jpg)
上QQ阅读APP看书,第一时间看更新
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=1738954381-dCTLOo3ajvwkcEjXndxbqyeKWWigkcBO-0-bb22376872ba71320387edc7902c3034)
计算结果被完整打印出来,如图2.7所示。
![](https://epubservercos.yuewen.com/281CEB/23721624209516806/epubprivate/OEBPS/Images/Figure-P47_4421.jpg?sign=1738954381-Mrr1EZRBdkKI2q0eBy4AHY5nBogbkaHp-0-bb878d669f7c9495e01e0afc4c5d80d6)
图2.7 打印结果