Skip to content
GitLab
Select Git revision
  • master default
Find file BlameHistoryPermalink
utils.py 945 bytes 






1


2


3


4


5


6


7


8


9


10


11


12


13


14


15


16


17


18






import numpy as np

from extreme_fit.estimator.abstract_estimator import AbstractEstimator
from extreme_fit.model.margin_model.linear_margin_model.linear_margin_model import LinearMarginModel
from extreme_fit.function.margin_function.linear_margin_function import LinearMarginFunction


def load_margin_function(estimator: AbstractEstimator, margin_model: LinearMarginModel,
                         margin_function_class=LinearMarginFunction, coef_dict=None):
    if coef_dict is None:
        coef_dict = estimator.result_from_model_fit.margin_coef_ordered_dict
    return margin_function_class.from_coef_dict(coordinates=estimator.dataset.coordinates,
                                                param_name_to_dims=margin_model.margin_function_start_fit.param_name_to_dims,
                                                coef_dict=coef_dict,
                                                starting_point=margin_model.starting_point)