from ax.core.arm import Arm from ax.core.parameter import RangeParameter parameters = [RangeParameter('x', lower=0.0, upper=1.0)] parameterization = dict(zip([p.name for p in parameters], [p.to_json() for p in parameters])) arm = Arm(parameters=parameterization)
from ax.core.arm import Arm from ax.core.parameter import ChoiceParameter, ParameterType parameters = [ChoiceParameter(name='fruit', parameter_type=ParameterType.STRING, values=['apple', 'banana', 'orange'])] parameterization = dict(zip([p.name for p in parameters], [p.to_json() for p in parameters])) arm = Arm(parameters=parameterization)
from ax.core.arm import Arm from ax.core.parameter import RangeParameter parameters = [RangeParameter('x', lower=0.0, upper=1.0), RangeParameter('y', lower=0.0, upper=2.0)] parameterization1 = dict(zip([p.name for p in parameters], [{'value': 0.5}, {'value': 1.0}])) arm1 = Arm(parameters=parameterization1) parameterization2 = dict(zip([p.name for p in parameters], [{'value': 0.3}, {'value': 1.5}])) arm2 = Arm(parameters=parameterization2)In this example, we are defining two arms with same parameterization type but different parameter values. These two arms can be used in an experiment with other arms that have different parameterizations. The package library used in these examples is Ax platform's core package library.