|
|
@@ -133,8 +133,8 @@ class ConvGAN(GanBaseClass):
|
|
|
return synth_set
|
|
|
|
|
|
def predict(self, data):
|
|
|
- prediction = self.generator.predict(data)
|
|
|
- return np.array(map(lambda x: x[0], prediction))
|
|
|
+ prediction = self.maj_min_discriminator.predict(data)
|
|
|
+ return np.array([x[0] for x in prediction])
|
|
|
|
|
|
# ###############################################################
|
|
|
# Hidden internal functions
|
