add meta_learn integration

SimonBlanke · Oct 5, 2019 · 594ac63 · 594ac63
1 parent c5b2f00
commit 594ac63
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/hyperactive/base_optimizer.py b/hyperactive/base_optimizer.py
@@ -10,6 +10,7 @@
 
 from .base_positioner import BasePositioner
 from .util import initialize_search, finish_search_, sort_for_best
+from meta_learn import HyperactiveWrapper
 
 
 class BaseOptimizer:
@@ -56,6 +57,9 @@ def __init__(self, _config_, _arg_):
         self.search_config = self._config_.search_config
         self.n_iter = self._config_.n_iter
 
+        if self._config_.meta_learn:
+            self._meta_ = HyperactiveWrapper(self._config_.search_config)
+
         if self._config_.get_search_path:
             self.pos_list = []
             self.score_list = []
@@ -144,6 +148,11 @@ def _run_one_job(self, X, y):
             print("\nscore       =", self.score_best)
             print("start_point =", start_point)
 
+        print([_cand_])
+
+        if self._config_.meta_learn:
+            self._meta_.collect(X, y, _cand_list=[_cand_])
+
     def _run_multiple_jobs(self, X, y):
         _cand_list = self._search_multiprocessing(X, y)