Update

programs/regression_with_neural_network.py

@@ -11,6 +11,8 @@ def FrankeFunction(x, y):
     return term1 + term2 + term3 + term4
 
 def prepare_data(x, y, z):
+    """Prepare data on a form that the neural network can handle.
+       Only works on scalar functions of two variables, e.g. the Frankefunction."""
     assert len(x) == len(y) == len(z), "shape of x, y, z do not match"
     input_data = [np.array([i, j]) for i, j in zip(x, y)]
     for arr in input_data:
@@ -39,7 +41,6 @@ def prepare_data(x, y, z):
 net = NeuralNetwork(layers=layers, mode='regression')
 net.SGD(training_data, epochs=30, mini_batch_size=10, eta=3, lmbda=0)
 
-
 ztilde  = net.feedforward(input_test)
 ztilde  = ztilde.flatten()
 z_exact = z_test.flatten()