Model Layers - AI

Convolution Neural Net (CNN)

model.add(tf.keras.layers.Conv2D(32,
                                 (config.first_layer_conv_width,
                                  config.first_layer_conv_height),
                                 input_shape=(28, 28, 1),
                                 activation='relu'))
model.add(tf.keras.layers.MaxPooling2D(pool_size=(2, 2)))
model.add(tf.keras.layers.Flatten())
model.add(tf.keras.layers.Dense(config.dense_layer_size, activation='relu'))
model.add(tf.keras.layers.Dense(num_classes, activation='softmax'))