如何使整数输出层？

Question

我有这样的网络，我想让最后一层返回整数（目标数据是整数）

connection=[
    layers.Input(4),
    layers.Sigmoid(50),layers.Sigmoid(50),
    layers.Sigmoid(1),
],

Answer 1

您不能在训练期间进行此操作，因为您将需要取整您的值，并且它将完全消失梯度的信息，但是您可以做出非常接近整数的预测，并且当您需要使用它时，您可以将值转换为整数

from neupy import algorithms, layers

network = algorithms.GradientDescent([
    layers.Input(4),
    layers.Relu(50),
    layers.Relu(50),
    layers.Relu(1), 
])
prediction = network.predict(data).round()