Kristian Schultz 3 лет назад
Родитель
Сommit
34e27d0492
2 измененных файлов с 2 добавлено и 1 удалено
  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]
             nMinoryPoints = dataSet.data1.shape[0]
             if self.nebInitial is None:
             if self.nebInitial is None:
                 self.neb = nMinoryPoints
                 self.neb = nMinoryPoints
-            else
+            else:
                 self.neb = min(self.nebInitial, nMinoryPoints)
                 self.neb = min(self.nebInitial, nMinoryPoints)
         else:
         else:
             self.neb = self.nebInitial
             self.neb = self.nebInitial

+ 1 - 0
library/generators/ctab.py

@@ -17,6 +17,7 @@ class CtabGan(GanBaseClass):
     def __init__(self, epochs=10, debug=True):
     def __init__(self, epochs=10, debug=True):
         self.isTrained = False
         self.isTrained = False
         self.epochs = epochs
         self.epochs = epochs
+        self.canPredict = False
 
 
     def reset(self, _dataSet):
     def reset(self, _dataSet):
         """
         """