Parcourir la source

Fixed typo / missing value.

Kristian Schultz il y a 3 ans
Parent
commit
34e27d0492
2 fichiers modifiés avec 2 ajouts et 1 suppressions
  1. 1 1
      library/generators/convGAN.py
  2. 1 0
      library/generators/ctab.py

+ 1 - 1
library/generators/convGAN.py

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

+ 1 - 0
library/generators/ctab.py

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