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

问题描述:

使用keras 的Sequential 模型出现的问题。
AttributeError: ‘Sequential‘ object has no attribute ‘predict_classes‘

解决:

我们只能利用predict()得到预测概率。那么这种情况我们如何获取预测标签呢?
1,对于多分类模型, 或者输出层使用softmax作为激活函数的的二分类模型

pred = np.argmax(model.predict(X_test),axis=-1)

2,对于输出层使用sigmoid 激活函数的二分类

pred = np.int64(model.predict(X_test) > 0.5)

你可能感兴趣的:(错误,深度学习,人工智能)