hyperactive.experiment.toy._parabola

Complexity

Total Complexity

3

Size/Duplication

Total Lines	33
Duplicated Lines	0 %

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	Parabola.__init__()	0	5	1
A	Parabola._paramnames()	0	2	1
A	Parabola._score()	0	5	1

"""2D parabola."""

from hyperactive.base import BaseExperiment


class Parabola(BaseExperiment):
    """2D parabola.

    Parameters
    ----------
    a : float, default=1.0
        Coefficient of the parabola.
    b : float, default=0.0
        Coefficient of the parabola.
    c : float, default=0.0
        Coefficient of the parabola.
    """

    def __init__(self, a=1.0, b=0.0, c=0.0):
        self.a = a
        self.b = b
        self.c = c
        super().__init__()

    def _paramnames(self):
        return ["x", "y"]

    def _score(self, params):
        x = params["x"]
        y = params["y"]

        return self.a * (x**2 + y**2) + self.b * x + self.c * y, {}