소스 검색

Fixesd Optimizer parameter namings.

Kristian Schultz 4 년 전
부모
커밋
f5a1a079c4
2개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  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
     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

+ 1 - 1
library/testers.py

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