@@ -39,7 +39,7 @@ class SimpleGan(GanBaseClass):
@staticmethod
def _adamOptimizer():
- return Adam(lr=0.0002, beta_1=0.5)
+ return Adam(learning_rate=0.0002, beta_1=0.5)
def _createGan(self, noiseSize=100):
self.discriminator.trainable=False
@@ -158,7 +158,7 @@ def lr(ttd):
logreg = LogisticRegression(
C=1e5,
solver='lbfgs',
- max_iter=1000,
+ max_iter=10000,
multi_class='multinomial',
class_weight={0: 1, 1: 1.3}
)