from tensorflow.keras import Sequential, Input, Model
from tensorflow.keras.layers import Dense, LeakyReLU, BatchNormalization


def build_encoder(latent_dim):
    model = Sequential()

    model.add(Dense(64, input_dim=121, kernel_initializer='glorot_uniform'))
    model.add(LeakyReLU(alpha=0.1))
    model.add(BatchNormalization(momentum=0.8))
    model.add(Dense(latent_dim, kernel_initializer='glorot_uniform'))

    model.summary()

    real = Input(shape=(121,))
    z = model(real)

    return Model(real, z)