Как сделать целочисленный выходной слой?

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