rename

pinky/src/hoptimizer.py → pinky/src/optimize.py

@@ -9,16 +9,20 @@ class Optimizer(Object):
 
     path_best = String.T(default='winner')
 
-    def evaluate(self, model):
-        result = model.train()
-        loss = result['loss']
-        return loss
+    def __init__(self, **kwargs):
+        super().__init__(kwargs)
+        self.model = None
 
-    def optimize(self):
+    def evaluate(self, **kwargs):
+        ''' wrapper to parse gp_minimize args to model.train'''
+        return self.model.train(**kwargs)['loss']
+
+    def optimize(self, model):
+        self.model = model
         default_parameters = [
             self.learning_rate[-1]
             ]
-        gp_minimize(func=self.evaluate, x0=default_parameters)
+        gp_minimize(func=model.train, x0=default_parameters)
 
     def log_dir_name(self, learning_rate):