| from tensorflow.keras import layers | |
| from tensorflow.keras import models | |
| model = models.Sequential() | |
| model.add(layers.Conv2D(32, (3, 3), activation='relu', input_shape=(28, 28, 1))) | |
| model.add(layers.MaxPooling2D((2, 2))) | |
| model.add(layers.Conv2D(64, (3, 3), activation='relu')) | |
| model.add(layers.MaxPooling2D((2, 2))) | |
| model.add(layers.Conv2D(64, (3, 3), activation='relu')) | |
| model.add(layers.Flatten()) | |
| model.add(layers.Dense(64, activation='relu')) | |
| model.add(layers.Dense(10, activation='softmax')) | |
| model.summary() | |