Here, we just extend keras.Model and initialize our WGAN with a generator, discriminator, and related parameters.
generatoranddiscriminatorare the models we defined earlier to use in our GAN.discriminator_stepsandgenerator_stepsrepresent the number of steps to train the discriminator and generator in each iteration.gp_weightis the weight assigned to the gradient penalty term.