tenforflow调用训练的模型

学习目标：

tenforflow调用训练的模型

---

学习产出：

#@Time   :2021/01/04
#@Note   :调用之前的模型
import tensorflow as tf
import numpy as np
from sklearn.datasets import load_iris

data = load_iris()
iris_data = np.float32(data.data)
iris_target = data.target
iris_target = np.float32(tf.keras.utils.to_categorical(iris_target,num_classes=3))
new_model = tf.keras.models.load_model('the_save_model.h5')
new_prediction = new_model.predict(iris_data)
print(tf.argmax(new_prediction,axis = -1))