diff --git a/ML/TensorFlow/Basics/tutorial3-neuralnetwork.py b/ML/TensorFlow/Basics/tutorial3-neuralnetwork.py index a36a3d95..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),