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获取班级的概率，并从那里获取班级。