keras - Keras Dense 中的 output_dim 是什么意思？

Question

我是深度学习的新手。通过研究这篇文章 链接。

我无法理解output_dim=128和是什么output_dim=1意思？？？

我希望作为最终密集的输出，类数=2（猫/狗）。

除了128从哪里来？？？

标签： keras

Answer 1

output_dim 是密集嵌入的维度。

128的选择

classifier.add(Dense(output_dim = 128, activation = 'relu'))

很随意，它只是表示您喜欢的全连接层的大小。您可以将其更改为另一个数字。

1在

classifier.add(Dense(output_dim = 1, activation = 'sigmoid'))

这是由于对于二元分类问题，我们只需要一个概率来区分 2 个组。如果概率至少为 0.5，我们将其归类为狗，如果小于 0.5，我们将其归类为猫。

如果您愿意，您还可以将激活函数设置为 softmax 并将 output_dim 设置为 2 作为最后一层，但这不会提高性能。