|
|
@@ -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):
|
