Devanshchowdhury2212

def get_model():
    keras.backend.clear_session()
    model = keras.models.Sequential()
    model.add(keras.layers.Conv2D(32, (3, 3), activation='relu', input_shape=(268, 650, 1),padding='same'))
    model.add(keras.layers.MaxPooling2D((2, 2),padding='same'))
    model.add(keras.layers.Conv2D(32, (3, 3), activation='relu',padding='same'))
    model.add(keras.layers.MaxPooling2D((2, 2),padding='same'))
    model.add(keras.layers.Conv2D(64, (3, 3), activation='relu',padding='same'))
    model.add(keras.layers.MaxPooling2D((2, 2),padding='same'))
    model.add(keras.layers.Conv2D(64, (3, 3), activation='relu',padding='same'))

Devanshchowdhury2212

model.evaluate(train_features, train_labels)
model.evaluate(validation_features, validation_labels)
model.evaluate(test_features, test_labels)