Przeglądaj źródła

Fixesd Optimizer parameter namings.

Kristian Schultz 4 lat temu
rodzic
commit
f5a1a079c4
2 zmienionych plików z 2 dodań i 2 usunięć
  1. 1 1
      library/SimpleGan.py
  2. 1 1
      library/testers.py

+ 1 - 1
library/SimpleGan.py

@@ -39,7 +39,7 @@ class SimpleGan(GanBaseClass):
 
 
     @staticmethod
     @staticmethod
     def _adamOptimizer():
     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):
     def _createGan(self, noiseSize=100):
         self.discriminator.trainable=False
         self.discriminator.trainable=False

+ 1 - 1
library/testers.py

@@ -158,7 +158,7 @@ def lr(ttd):
     logreg = LogisticRegression(
     logreg = LogisticRegression(
         C=1e5,
         C=1e5,
         solver='lbfgs',
         solver='lbfgs',
-        max_iter=1000,
+        max_iter=10000,
         multi_class='multinomial',
         multi_class='multinomial',
         class_weight={0: 1, 1: 1.3}
         class_weight={0: 1, 1: 1.3}
         )
         )