如何使神经网络输出一定范围内的特定数目？

Question

我有一个神经网络，一个输入层，两个隐藏的密集层和一个输出层。

对于输出层中的每个神经元，我希望给出一个介于0到2之间的数字(0、1或2)，如下所示：

如果我对每一种可能性使用一个神经元(一个神经元为0，一个神经元为1，另一个神经元为2)，然后选择一个具有最佳预测值的神经元，输出层的长度就太长了。

有没有办法实现这一点？(我对神经网络等很陌生)

Answer 1

有两种方法可以做到这一点：

1. 将输出数据从0，2缩放到0，1，并在最后应用Sigmoid激活。
2. 创建自己的自定义激活函数，该函数输出0，2中的所有内容

我强烈建议你第一，这是更快的实现。