adviserastar.py

'''
Copyright (C) e2its - All Rights Reserved
 * Unauthorized copying of this file, via any medium is strictly prohibited
 * Proprietary and confidential
 *
 * This file is part of gDayF project.
 *
 * Written by Jose L. Sanchez <e2its.es@gmail.com>, 2016-2019
'''


from gdayf.core.adviserbase import Adviser
from gdayf.common.utils import get_model_fw
import importlib
''' Don't delete used for dynamic optimizer methods'''
from gdayf.conf.loadconfig import LoadConfig
from gdayf.common.utils import decode_ordered_dict_to_dataframe
from gdayf.models.parametersmetadata import ParameterMetadata



class AdviserAStar(Adviser):

    
    def __init__(self, e_c, deep_impact=3, metric='train_accuracy', dataframe_name='', hash_dataframe=''):
        super(AdviserAStar, self).__init__(e_c=e_c, deep_impact=deep_impact, metric=metric,
                                           dataframe_name=dataframe_name, hash_dataframe=hash_dataframe)

    
    def optimize_models(self, armetadata):
        metric_value, _, objective = eval('self.get_' + self.metric + '(armetadata)')
        engine = get_model_fw(armetadata)
        for engines in [*self._frameworks]:
            if engine == engines:
                optimizer_engine = importlib.import_module(self._frameworks[engine]['conf']['optimization_method'])
                model_list = optimizer_engine.Optimizer(self._ec).optimize_models(armetadata=armetadata,
                                                                                  metric_value=metric_value,
                                                                                  objective=objective,
                                                                                  deepness=self.deepness,
                                                                                  deep_impact=self.deep_impact)
                optimized_model_list = list()
                for model in model_list:
                    self.safe_append(optimized_model_list, model)

                return (optimized_model_list)