代码之家  ›  专栏  ›  技术社区  ›  Tony Ennis

一个输出,7个可能的值。什么是正确的损失函数?

  •  1
  • Tony Ennis  · 技术社区  · 3 年前

    我的CNN在训练时的准确度不会超过0.14。奇怪的是,这个模型运行得很好。对于具有单节点输出且输出值应为1到7(包括1和7)之间的整数的CNN,正确的损耗函数是什么?(输出是一个类别。)

    (我忍不住注意到.14xx*7大约是1。)

    此外,我的单节点输出上的激活函数应该是什么?我没有具体说明。

    0 回复  |  直到 3 年前
        1
  •  1
  •   I'mahdi    3 年前

    你正在做一个分类问题,有7个类。因此,模型中的最后一层应该是

    output=Dense(units=7, activation='softmax') (x)
    

    在你的模型中,使用损失作为分类熵和标签编码 应该是明确的