Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit aa47df0

Browse files
committedSep 18, 2024·
cnn_mnist.py: For the Convolutional Neural Network (CNN) model.
1 parent 3a47c23 commit aa47df0

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed
 

‎machine_learning/neural_networks/cnn_mnist.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,8 @@
5757
model.add(layers.Dense(10, activation="softmax"))
5858

5959
# Compile the model
60-
model.compile(optimizer="adam", loss="categorical_crossentropy", metrics=["accuracy"])
60+
model.compile(optimizer="adam", loss="categorical_crossentropy",
61+
metrics=["accuracy"])
6162

6263
# Display the model summary
6364
model.summary()

0 commit comments

Comments
 (0)
Please sign in to comment.