diff --git a/ML/TensorFlow/Basics/tutorial3-neuralnetwork.py b/ML/TensorFlow/Basics/tutorial3-neuralnetwork.py index adaf13bb..208623f3 100644 --- a/ML/TensorFlow/Basics/tutorial3-neuralnetwork.py +++ b/ML/TensorFlow/Basics/tutorial3-neuralnetwork.py @@ -17,7 +17,7 @@ # Sequential API (Very convenient, not very flexible) model = keras.Sequential( [ - keras.Input(shape=(28 * 28)), + keras.Input(shape=(28 * 28,)), layers.Dense(512, activation="relu"), layers.Dense(256, activation="relu"), layers.Dense(10), @@ -39,7 +39,7 @@ model.compile( loss=keras.losses.SparseCategoricalCrossentropy(from_logits=False), - optimizer=keras.optimizers.Adam(lr=0.001), + optimizer=keras.optimizers.Adam(learning_rate=0.001), metrics=["accuracy"], )