Source code for ax.metrics.branin
#!/usr/bin/env python3
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
import numpy as np
from ax.metrics.noisy_function import NoisyFunctionMetric
from ax.utils.common.typeutils import checked_cast
from ax.utils.measurement.synthetic_functions import branin
[docs]class BraninMetric(NoisyFunctionMetric):
[docs] def f(self, x: np.ndarray) -> float:
x1, x2 = x
return checked_cast(float, branin(x1=x1, x2=x2))
[docs]class NegativeBraninMetric(BraninMetric):
[docs] def f(self, x: np.ndarray) -> float:
fpos = super().f(x)
return -fpos