Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- training.shape (50000, 28, 28)
- testing.shape (2938, 28, 28)
- model = Sequential()
- model.add(Conv2D(64, kernel_size=3, activation='relu', input_shape=(28,28,1)))
- model.add(Conv2D(32, kernel_size=3, activation='relu'))
- model.add(Flatten())
- model.add(Dense(10, activation='softmax'))
- Layer (type) Output Shape Param #
- =================================================================
- conv2d_1 (Conv2D) (None, 26, 26, 64) 640
- _________________________________________________________________
- conv2d_2 (Conv2D) (None, 24, 24, 32) 18464
- _________________________________________________________________
- flatten_1 (Flatten) (None, 18432) 0
- _________________________________________________________________
- dense_1 (Dense) (None, 10) 184330
- =================================================================
- Total params: 203,434
- Trainable params: 203,434
- Non-trainable params: 0
- _________________________________________________________________
- model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
Add Comment
Please, Sign In to add comment