ソースを参照

Fixed typo / missing value.

Kristian Schultz 3 年 前
コミット
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]
             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):
         """