AttributeError: 'Sequential' object 沒有屬性 'predict_classes'

Question

我構建了一個 CNN，可以對來自兩個不同類別的圖像進行分類。 但是，當我嘗試運行分類代碼時，我不斷收到此錯誤：

AttributeError: 'Sequential' object has no attribute 'predict_classes'

這是導致錯誤的代碼部分：

 images = np.vstack([x])
 classes = model.predict_classes(images, batch_size=10)

Answer 1

predict_classes方法來自舊版本的 keras（在版本 2 之前，因此 2017 年及更早版本）並從那時起被刪除。

您可以使用model.predict獲取班級的概率，並從那里獲取班級。