问题描述:
使用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)