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)
예제 #2
0
파일: plot_mds.py 프로젝트: noorawad/DE-NAS
        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/"