def specification(self) -> problem.ProblemSpec:
params = [
problem.Var("x", problem.ContinuousRange(-10, 10)),
problem.Var("y", problem.DiscreteRange(-3, 3)),
]
return problem.ProblemSpec(
name="Quadratic Function", params=params, values=[problem.Var("x**2 + y")]
)
def specification(self) -> problem.ProblemSpec:
params = [
problem.Var("x1", problem.ContinuousRange(-5, 10)),
problem.Var("x2", problem.ContinuousRange(-5, 10)),
]
return problem.ProblemSpec(
name="Rosenbrock Function",
params=params,
values=[problem.Var("Rosenbrock")],
)
def specification(self) -> problem.ProblemSpec:
params = [
problem.Var("x1", problem.ContinuousRange(-5, 10)),
problem.Var("x2", problem.ContinuousRange(-5, 10)),
]
return problem.ProblemSpec(
name="Six-Hemp Camel Function",
params=params,
values=[problem.Var("Six-Hemp Camel")],
)
def specification(self) -> problem.ProblemSpec:
params = [
problem.Var("x1", problem.ContinuousRange(-4, 4)),
problem.Var("x2", problem.ContinuousRange(-4, 4)),
]
return problem.ProblemSpec(
name="Himmelblau Function",
params=params,
values=[problem.Var("Himmelblau")],
)
def specification(self) -> problem.ProblemSpec:
params = [
problem.Var(f"x{i+1}", problem.ContinuousRange(-5.12, 5.12))
for i in range(self.dim)
]
return problem.ProblemSpec(
name=f"Rastrigin (dim={self.dim})",
params=params,
values=[problem.Var("Rastrigin")],
)
def specification(self) -> problem.ProblemSpec:
params = [
problem.Var(f"x{i}", problem.CategoricalRange(op_names))
for i in range(edge_num)
]
return problem.ProblemSpec(
name=f"NASBench201-{self._dataset}",
params=params,
values=[problem.Var("value")],
steps=list(range(prune_start_epoch, max_epoch, prune_epoch_step)) +
[max_epoch],
)
def specification(self) -> problem.ProblemSpec:
self._n_wfg = int(sys.argv[1])
self._n_dim = int(sys.argv[2])
self._low = 0
self._high = 2
params = [
problem.Var(f"x{i}", problem.ContinuousRange(0, self._high * i))
for i in range(self._n_dim)
]
return problem.ProblemSpec(
name=f"WFG{self._n_wfg}",
params=params,
values=[problem.Var("f1"), problem.Var("f2")],
)
