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


def build_discriminator():
    model = Sequential()
    model.add(Dense(256, input_dim=121, kernel_initializer='glorot_uniform'))
    model.add(LeakyReLU(alpha=0.1))
    model.add(Dropout(0.2))
    model.add(Dense(128, kernel_initializer='glorot_uniform'))
    model.add(LeakyReLU(alpha=0.1))
    model.add(Dropout(0.2))
    model.add(Dense(128, kernel_initializer='glorot_uniform'))
    model.add(LeakyReLU(alpha=0.1))
    model.add(Dropout(0.2))
    model.add(Dense(1, kernel_initializer='glorot_uniform', activation='sigmoid'))
    model.summary()

    data = Input(shape=(121,))
    validity = model(data)
    return Model(data, validity)