def back_prop_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]]) forward_res = net.forward_prop(x) res = net.back_prop(1, forward_res)