编码时出现错误：“AttributeError: 'Sequential' object has no attribute 'predict_classes'”。 这里错了

Question

def classify(file_path):
  global label_packed
  image = Image.open(file_path)
  image = image.resize((30,30))
  image = numpy.expand_dims(image, axis=0)
  image = numpy.array(image)
  pred = model.predict_classes(image)[0]
  sign = classes[pred+1]
  print(sign)
  label.configure(foreground='#011638', text=sign) 

Answer 1

更改：

model.predict_classes() 

至

model.predict() 

它会返回一个概率。 然后你应该定义一个阈值来考虑对其中一个类的预测概率。 请记住，阈值取决于您。 您可以对此严格或不严格。 但是，我认为它是 0.5：

pred_class = np.where(model.predict(example) > 0.5, 1, 0)

然后，您将获得预测的类。