[英]How can I see the model predicted answer and real output using model(.h5)?
我使用 CNN 制作了簡單的圖像分類模型(classification.h5),我希望看看我的 model 是否工作正常。
我的 CNN model 是:
from tensorflow.keras import layers
from tensorflow.keras.models import Sequential
model = Sequential()
model.add(Conv2D(16, (3, 3), activation='relu', strides=(1, 1),
padding='same',input_shape=input_shape))
model.add(MaxPool2D((2, 2)))
model.add(Dropout(0.5))
model.add(Conv2D(32, (3, 3), activation='relu', strides=(1, 1),
padding='same'))
model.add(MaxPool2D((2, 2)))
model.add(Dropout(0.5))
model.add(Conv2D(412, (13, 13), strides=(1, 1), padding = 'same', activation='relu'))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Dropout(0.5))
model.add(Flatten())
model.add(Dense(128, activation='relu'))
model.add(Dense(64, activation='relu'))
model.add(Dense(4, activation='softmax'))
model.compile(loss='categorical_crossentropy', optimizer='adam',metrics=['acc'])
model 運行良好。 對於下一步,我試圖看看這個 model 是否真的運作良好,但我迷失了如何才能看到真正的結果。
我想嘗試查看的內容如下所示:
model prediction = '模型預測答案'
真實答案 = '真實答案'
我該如何編寫這個 output 的代碼?
saved_model = keras.models.load_model("Model_Name.h5")
准備數據(預處理或重塑)
做你的預測:
saved_model.predict(data)
注意:您可以像這樣保存 model: model.save("Model_Name.h5")
要查看您的 model 是否真的有效,您必須測試未在訓練中使用的數據的預測,因此您必須執行類似的操作
預測值 = model.predict(data_not_used_in_training)
將這些預測值與該數據的實際值進行比較並應用指標
查看有關如何進行預測的鏈接
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.