@@ -63,7 +63,7 @@ class ConvGAN(GanBaseClass):
nMinoryPoints = dataSet.data1.shape[0]
if self.nebInitial is None:
self.neb = nMinoryPoints
- else
+ else:
self.neb = min(self.nebInitial, nMinoryPoints)
else:
self.neb = self.nebInitial
@@ -17,6 +17,7 @@ class CtabGan(GanBaseClass):
def __init__(self, epochs=10, debug=True):
self.isTrained = False
self.epochs = epochs
+ self.canPredict = False
def reset(self, _dataSet):
"""