neupy - 如何使整数输出层？

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()