Discriminator.py 765 B

123456789101112131415161718192021
  1. from tensorflow.keras import Input, Model
  2. from tensorflow.keras.layers import concatenate, Dense, LeakyReLU, Dropout
  3. def build_discriminator(latent_dim):
  4. z = Input(shape=(latent_dim,))
  5. data = Input(shape=(121,))
  6. d_in = concatenate([z, data])
  7. model = Dense(128, kernel_initializer='glorot_uniform')(d_in)
  8. model = LeakyReLU(alpha=0.1)(model)
  9. model = Dropout(0.2)(model)
  10. model = Dense(128, kernel_initializer='glorot_uniform')(model)
  11. model = LeakyReLU(alpha=0.1)(model)
  12. model = Dropout(0.2)(model)
  13. model = Dense(128, kernel_initializer='glorot_uniform')(model)
  14. model = LeakyReLU(alpha=0.1)(model)
  15. model = Dropout(0.2)(model)
  16. validity = Dense(1, activation="sigmoid")(model)
  17. return Model([z, data], validity)