Esempio n. 1
0
 def calculate_gradient_test(self):
     net = NeuralNet([1, 3, 1], -1, 1)
     net_layers = net.return_net()
     net_layers[0].set_matrix(numpy.array([[1, 1], [2, 2], [3, 3]]))
     net_layers[1].set_matrix(numpy.array([1, 2, 3, 4]))
     net_layers[1].set_matrix(
         numpy.reshape(net_layers[1].return_matrix(), (1, 4)))
     net.set_net(net_layers)
     x = numpy.array([[-3]])
     res = net.calculate_gradients([[x], [1]])