[ANTBO]: add option to optimize from table of candidates.

qualtric · Mar 10, 2024 · e7f18bd · e7f18bd
1 parent e0dfd88
commit e7f18bd
Show file tree

Hide file tree

Showing 10 changed files with 569 additions and 496 deletions.
diff --git a/AntBO/bo/botask.py b/AntBO/bo/botask.py
@@ -1,6 +1,6 @@
 import numpy as np
 from bo.base import TestFunction
-from task.tools import Absolut, Manual
+from task.tools import Absolut, Manual, TableFilling
 import torch
 
 class BOTask(TestFunction):
@@ -28,6 +28,8 @@ def __init__(self,
             self.fbox = Absolut(self.bbox)
         elif self.bbox['tool'] == 'manual':
             self.fbox = Manual(self.bbox)
+        elif self.bbox['tool'] == 'table_filling':
+            self.fbox = TableFilling(self.bbox)
         else:
             assert 0,f"{self.bbox['tool']} Not Implemented"
 

diff --git a/AntBO/bo/localbo_cat.py b/AntBO/bo/localbo_cat.py