class TestFCNetProteinStructure(unittest.TestCase): def setUp(self): self.b = FCNetProteinStructureBenchmark( data_dir="./fcnet_tabular_benchmarks/") def test_random_sampling(self): config = self.b.get_configuration_space().sample_configuration() self.b.objective_function(config)
b = FCNetSliceLocalizationBenchmark(data_dir=data_dir) elif benchmark_type == "naval_propulsion": # NAS-HPO-Bench max_budget = 100 b = FCNetNavalPropulsionBenchmark(data_dir=data_dir) else: # benchmark_type == "parkinsons_telemonitoring": # NAS-HPO-Bench max_budget = 100 b = FCNetParkinsonsTelemonitoringBenchmark(data_dir=data_dir) def f(config, budget=None): if budget is not None: fitness, cost = b.objective_function(config, budget=int(budget)) else: fitness, cost = b.objective_function(config) return fitness, cost cs = b.get_configuration_space() dimensions = len(cs.get_hyperparameters()) elif benchmark == '1shot1': assert benchmark_type in ['1', '2', '3'] sys.path.append(os.path.join(os.getcwd(), '../nasbench/')) sys.path.append(os.path.join(os.getcwd(), '../nasbench-1shot1/')) from nasbench import api from nasbench_analysis.search_spaces.search_space_1 import SearchSpace1 from nasbench_analysis.search_spaces.search_space_2 import SearchSpace2 from nasbench_analysis.search_spaces.search_space_3 import SearchSpace3 nasbench = api.NASBench( os.path.join( os.getcwd(), "../nasbench-1shot1/nasbench_analysis/"