objective_function.py

# coding=utf-8
import series_utilities as util
import numpy as np

PENALIZATION_VALUE = 10000


def minimize_slope_std(a_series, *args):
    cuts_list, minmax_list = args

    a_series_split = util.split_series_for_dimension(a_series, cuts_list)
    trap_series_split = util.rebuild_series(a_series_split, cuts_list, minmax_list)

    compute_slope_fun = util.arctan_segment_slope
    slope_multi_list = [util.get_slope_list(trap_series, compute_slope_fun) for trap_series in trap_series_split]
    slope_std_list = [np.std(slope_list) for slope_list in slope_multi_list]

    result = sum(slope_std_list) / len(slope_std_list)
    return result


def minimize_slope_sum(a_series, *args):
    cuts_list, minmax_list = args

    a_series_split = util.split_series_for_dimension(a_series, cuts_list)
    trap_series_split = util.rebuild_series(a_series_split, cuts_list, minmax_list)

    compute_slope_fun = util.arctan_segment_slope
    slope_multi_list = [util.get_slope_list(trap_series, compute_slope_fun) for trap_series in trap_series_split]

    """
    for slope_list in slope_multi_list:
        if _rect_instead_trap(slope_list):
            return PENALIZATION_VALUE
    """
    slope_sum_list = [sum(slope_list) for slope_list in slope_multi_list]

    result = sum(slope_sum_list) / len(slope_sum_list)
    return -result