Discriminator.py 761 B

123456789101112131415161718192021
  1. from tensorflow.keras import Sequential, Input, Model
  2. from tensorflow.keras.layers import Dense, LeakyReLU, Dropout
  3. def build_discriminator():
  4. model = Sequential()
  5. model.add(Dense(256, input_dim=121, kernel_initializer='glorot_uniform'))
  6. model.add(LeakyReLU(alpha=0.1))
  7. model.add(Dropout(0.2))
  8. model.add(Dense(128, kernel_initializer='glorot_uniform'))
  9. model.add(LeakyReLU(alpha=0.1))
  10. model.add(Dropout(0.2))
  11. model.add(Dense(128, kernel_initializer='glorot_uniform'))
  12. model.add(LeakyReLU(alpha=0.1))
  13. model.add(Dropout(0.2))
  14. model.add(Dense(1, kernel_initializer='glorot_uniform', activation='sigmoid'))
  15. model.summary()
  16. data = Input(shape=(121,))
  17. validity = model(data)
  18. return Model(data, validity)