Keras深度学习:入门、实战与进阶
上QQ阅读APP看书,第一时间看更新

3.1.6 做出预测

一旦我们对训练模型的性能感到满意,就可以用它来预测新数据。调用模型的predict()函数实现。

> predictions <- model %>% predict(x_test_scale,verbose = 0)
> head(predictions)
          [,1]
[1,] 0.9309239
[2,] 0.2624388
[3,] 0.9582632
[4,] 0.9527556
[5,] 0.6007361
[6,] 0.5770388

预测输出的形式由网络输出层的格式决定。在回归问题上,输出结果就是样本预测值,一般由线性激活函数完成。对于二元分类问题,预测结果以其中一个类别的概率形式出现。我们从预测结果前六行可知,返回结果为每个样本预测为1的概率值。

对于分类问题,我们还可以使用predict_classes()函数,该函数会自动将预测转换为概率值最大对应的类别值。

> pred_label <- model %>% predict_classes(x_test_scale,verbose = 0)
> head(pred_label)
     [,1]
[1,]    1
[2,]    0
[3,]    1
[4,]    1
[5,]    1
[6,]    1