magnusbarata

# %%
from tensorflow import keras
import numpy as np

def base_model(image_shape, n_class):
    input_image = keras.Input(shape=image_shape, dtype='float')
    x = keras.layers.Conv2D(32, kernel_size=3, activation='relu')(input_image)
    x = keras.layers.MaxPool2D(pool_size=2)(x)
    x = keras.layers.Conv2D(64, kernel_size=3, activation='relu')(x)
    x = keras.layers.MaxPool2D(pool_size=2)(x)