TensorFlow知识图谱实战
上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】

计算结果被完整打印出来,如图2.7所示。

图2.7 打印结果