我尝试使用 sigmoid 和 relu 函数来实现一个简单的神经网络。使用 sigmoid 函数,我得到了一些不错的输出。但是当使用 relu 时,我得到了 0 或 1 的数组。(我需要 relu 函数,因为我愿意将代码用于某些输出>1)。
def relu(x):
return np.maximum(0,x)
def reluDerivative(x):
x[x<=0] = 0
x[x>0] = 1
return x
training_inputs = np.array([[9, 0 , 1],
[7, 1, 1],
[8, 0, 1],
[5, 1, 1]
])
training_outputs = np.array([[9, 7, 8, 5]]).T
np.random.seed(1)
synaptic_weights = 2 * np.random.random((3,1)) - 1
for iteration in range(100000):
outputs = relu(np.dot(training_inputs, synaptic_weights))
error = training_outputs - outputs
adjustments = error * reluDerivative(outputs)
synaptic_weights += np.dot(training_inputs.T, adjustments )
print("output after training: \n" , outputs)