@@ -35,7 +35,7 @@ def repeat(x, times):
def create01Labels(totalSize, sizeFirstHalf):
labels = repeat(np.array([1,0]), sizeFirstHalf)
- labels.extend(repeat(np.array([0,1]), totalSize))
+ labels.extend(repeat(np.array([0,1]), totalSize - sizeFirstHalf))
return np.array(labels)
class ConvGAN(GanBaseClass):
