Skip to content

Commit 8484141

Browse files
committed
update AC-discovery comments
1 parent 4050a7f commit 8484141

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/AC-discovery.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,13 +51,13 @@ def f_model(u_model, var, x, t):
5151
# append to a list for input to model.fit
5252
X = [x, t]
5353

54-
# define col_weights for SA discovery model
54+
# define col_weights for SA discovery model, can be removed
5555
col_weights = tf.Variable(tf.random.uniform([np.shape(x)[0], 1]))
5656

5757
# initialize, compile, train model
5858
model = DiscoveryModel()
5959
model.compile(layer_sizes, f_model, X, u_star, params,
60-
col_weights=col_weights) # baseline approach can be done by simply removing the col_weights arg
60+
col_weights=col_weights) # baseline discovery approach can be done by simply removing the col_weights arg
6161
model.tf_optimizer_weights = tf.keras.optimizers.Adam(lr=0.005,
6262
beta_1=.95) # an example as to how one could modify an optimizer, in this case the col_weights optimizer
6363

0 commit comments

Comments
 (0)