@TOC
问题原因:在Sequential的网络架构中并没有predict_classes方法
解决
虽然没有给出predict_classes方法,但在Sequential中有predict方法,我们可以得到对于不同类别分类的概率
prediction = network.predict(x_test)
network是我们构建的Sequential网络,x_test是测试集的数据部分
在这里我们得到的prediction实际上是一个二维矩阵,它的维度是(len(x_test), 类别数)
我们希望将所有预测的结果放在prediction中(单个预测结果指的是对应最大概率中的类别)
(1)对于二分类问题,以0.5为界(IMDB数据集分类)
假设对应概率大于0.5为正例,否则为负例
这里二分类得到的是列数为1的矩阵
我们首先将prediction转换为一个数组,prediction[i]表示第i个样本对应的分类
为了方便区分,我们用results表示得到的一维数组
results = [