【发布时间】:2021-07-27 15:26:07
【问题描述】:
我有一个二元交叉熵模型,分为两类:[正常,看涨]。
当我调用 model.predict() 时,输出返回一个概率,我怎么知道概率属于哪个类?
我使用的是 keras 函数模型。
final layer: 1 unit, sigmoid activation.
classes: ["normal", "bullish"]
class indices: {'bullish': 0, 'normal': 1}
那么当单个预测返回 [0.5905021] 时,它有 59% 的机会是看涨还是正常类别?
【问题讨论】:
标签: python python-3.x keras deep-learning