簡體   English   中英

如何使用模型(.h5)查看 model 預測答案和真實 output?

[英]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 的代碼?

  1. 上傳保存的 model:
saved_model = keras.models.load_model("Model_Name.h5")
  1. 准備數據(預處理或重塑)

  2. 做你的預測:

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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM