神经网络输出预测及其概率

Posted 2021-04-25

有没有办法训练神经网络输出两个值 - 预测及其概率？

例如：假设我们想要预测时间序列1, 2, 3, 4, 5, ?，我们想知道两个数字 - 预测本身及其概率 - 神经网络对其预测的确定程度（在这种情况下它可能是两个数字6和90%）。

你知道这是否可能？以及对神经网络的文档/示例的任何参考，或那些类似的东西？

注意：预测值不是分类/类，它是一个数字。

答案

我不确定你使用什么库来训练神经网络，但通常它们能够产生预测概率和预测类。

以keras为例，训练有素的模型有以下两种方法：

1. predict_proba：预测输出类概率
2. predict_classes：预测输出类指数

通常（2）是（1），阈值为0.5。您可以调用它们来获取两个输出预测，也可以调用（1）然后设置自定义阈值。