有没有办法训练神经网络输出两个值 - 预测及其概率?
示例:假设我们要预测时间序列1, 2, 3, 4, 5, ?
,我们想要知道两个数字 - 预测本身及其概率 - 神经网络对其预测的确定程度(在这种情况下它可能是两个数字) 6
和90%
)。
你知道这是否可能?以及对神经网络的文档/示例的任何参考,或者类似的东西?
注意:预测值不是分类/类,它是一个数字。
答案 0 :(得分:0)
我不确定您使用什么库来训练神经网络,但通常它们能够产生预测概率和预测类别。
以keras
为例,训练有素的模型有以下两种方法:
predict_proba
:预测输出类概率predict_classes
:预测输出类索引通常(2)是(1),阈值为0.5。您可以调用它们来获得两个输出预测,也可以调用(1)然后设置自定义阈值。