20.6 做预测
问题描述
使用神经网络做预测。
解决方案
使用Keras构建一个前馈神经网络,接着使用predict方法来做预测:
讨论
通过Keras做预测是一件很容易的事。一旦训练完神经网络,接着就可以使用predict方法了,它把一组特征当作输入参数,返回对每个观察值的预测值。在解决方案中,我们的神经网络被设定为做二元分类的预测,所以输出是预测观察值为分类1的概率。如果对观察值预测的值非常接近1,则说明它很可能属于分类1;反之,如果预测的值非常接近0,那么观察值很可能属于分类0。下面的代码查看的是测试集中第一个观察值属于分类1的预测概率:
本书评论