Fix hyperopt space label error

kakao · Oct 13, 2020 · fbb1a14 · fbb1a14
1 parent d8af029
commit fbb1a14
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/buffalo/algo/options.py b/buffalo/algo/options.py
@@ -203,9 +203,9 @@ def get_default_optimize_option(self):
                 'd': ['randint', ['d', 10, 30]],
                 'reg_u': ['uniform', ['reg_u', 0.1, 1]],
                 'reg_i': ['uniform', ['reg_i', 0.1, 1]],
-                'reg_c': ['uniform', ['reg_i', 0.1, 1]],
+                'reg_c': ['uniform', ['reg_c', 0.1, 1]],
                 'alpha': ['randint', ['alpha', 1, 32]],
-                'l': ['randint', ['alpha', 1, 32]]
+                'l': ['randint', ['l', 1, 32]]
             }
         })
         return Option(opt)
@@ -366,7 +366,7 @@ def get_default_option(self):
         return Option(opt)
 
     def get_default_optimize_option(self):
-        """Optimization options for BPRMF.
+        """Optimization options for WARP.
         """
         opt = super().get_default_optimize_option()
         opt.update({
@@ -380,7 +380,7 @@ def get_default_optimize_option(self):
                 'threshold': ['uniform', ['threshold', 0.5, 5.0]],
                 'reg_u': ['uniform', ['reg_u', 0.01, 1.0]],
                 'reg_i': ['uniform', ['reg_i', 0.0, 0.001]],
-                'reg_j': ['uniform', ['reg_i', 0.0, 0.001]]
+                'reg_j': ['uniform', ['reg_j', 0.0, 0.001]]
             }
         })
         return Option(opt)