hyperactive.results_manager

Complexity

Total Complexity

5

Size/Duplication

Total Lines	52
Duplicated Lines	0 %

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	ResultsManagerMemory.load_long_term_memory()	0	2	1
A	ResultsManagerMemory.__init__()	0	13	1
A	ResultsManager.__init__()	0	5	1
A	ResultsManagerBase.__init__()	0	9	1
A	ResultsManagerMemory.save_long_term_memory()	0	2	1

# Author: Simon Blanke
# Email: simon.blanke@yahoo.com
# License: MIT License


from optimization_metadata import HyperactiveWrapper
from .meta_data.meta_data_path import meta_data_path


class ResultsManagerBase:
    def __init__(
        self, search_name, objective_function, search_space, function_parameter,
    ):
        self.search_name = search_name
        self.objective_function = objective_function
        self.search_space = search_space
        self.function_parameter = function_parameter

        self.memory_dict_new = {}


class ResultsManager(ResultsManagerBase):
    def __init__(
        self, search_name, objective_function, search_space, function_parameter,
    ):
        super().__init__(
            search_name, objective_function, search_space, function_parameter
        )


class ResultsManagerMemory(ResultsManagerBase):
    def __init__(
        self, search_name, objective_function, search_space, function_parameter,
    ):
        super().__init__(
            search_name, objective_function, search_space, function_parameter
        )

        self.hypermem = HyperactiveWrapper(
            main_path=meta_data_path(),
            X=function_parameter["features"],
            y=function_parameter["target"],
            model=self.objective_function,
            search_space=search_space,
        )

    def load_long_term_memory(self):
        return self.hypermem.load()

    def save_long_term_memory(self):
        self.hypermem.save(self.memory_dict_new)