modify init learning rate

niiboy · Apr 2, 2022 · 0dda232 · 0dda232
1 parent a8a3278
commit 0dda232
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/obp/policy/offline.py b/obp/policy/offline.py
@@ -743,7 +743,7 @@ class NNPolicyLearner(BaseOfflinePolicyLearner):
     solver: str = "adam"
     alpha: float = 0.0001
     batch_size: Union[int, str] = "auto"
-    learning_rate_init: float = 0.0001
+    learning_rate_init: float = 0.001
     max_iter: int = 200
     shuffle: bool = True
     random_state: Optional[int] = None
@@ -1477,7 +1477,7 @@ class QFuncEstimator:
     solver: str = "adam"
     alpha: float = 0.0001
     batch_size: Union[int, str] = "auto"
-    learning_rate_init: float = 0.0001
+    learning_rate_init: float = 0.001
     max_iter: int = 200
     shuffle: bool = True
     random_state: Optional[int] = None