update AC-discovery comments

Author: levimcclenny

examples/AC-discovery.py

@@ -35,9 +35,6 @@ def f_model(u_model, var, x, t):
 Exact = data['uu']
 Exact_u = np.real(Exact)
 
-# define MLP depth and layer width
-layer_sizes = [2, 128, 128, 128, 128, 1]
-
 # generate all combinations of x and t
 X, T = np.meshgrid(x, t)
 
@@ -47,13 +44,15 @@ def f_model(u_model, var, x, t):
 x = X_star[:, 0:1]
 t = X_star[:, 1:2]
 
-print(np.shape(x))
 # append to a list for input to model.fit
 X = [x, t]
 
 # define col_weights for SA discovery model, can be removed
 col_weights = tf.Variable(tf.random.uniform([np.shape(x)[0], 1]))
 
+# define MLP depth and layer width
+layer_sizes = [2, 128, 128, 128, 128, 1]
+
 # initialize, compile, train model
 model = DiscoveryModel()
 model.compile(layer_sizes, f_model, X, u_star, params,