tests.test_setup.experiment

Complexity

Total Complexity

2

Size/Duplication

Total Lines	12
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	SphereFunction.objective_function()	0	2	1
A	SphereFunction.setup()	0	2	1

import numpy as np

from hyperactive.experiment import BaseExperiment


class SphereFunction(BaseExperiment):
    def setup(self, n_dim=2):
        self.n_dim = n_dim

    def objective_function(self, params):
        return np.sum(params["x0"] ** 2 + params["x1"] ** 2)