例如,模型最终可能只是记住了训练样本和目标值之间的映射关
系,但这对在前所未见的数据上进行预测毫无用处。
与 MNIST 数据集一样, IMDB 数据集也内置于 Keras 库。...构建数据
将我们得到的数据构建成可以被神经网络处理的数据,也就是张量数据.需要通过一些方法:
填充列表,使其具有相同的长度,再将列表转换成形状为 (samples, word_indices)的整数张量...网络可以描述为;
本次使用顺序模型进行编程.
Keras构建网络
在 Keras 中有两类主要的模型:Sequential 顺序模型 和 使用函数式 API 的 Model 类模型。...由于面对的是一个二分类问题,网络输出是一个概率值(网络最后一层使用 sigmoid 激活函数,仅包含一个单元),那么最好使用 binary_crossentropy (二元交叉熵)损失。...当然这不是唯一可行的选择,比如还可以使用 mean_squared_error (均方误差)。但对于输出概率值的模型,交叉熵(crossentropy)往往是最好的选择。