def generate_continous_predicates()

in rules/predicate.py [0:0]

• 12 lines of code
• 4 McCabe index (conditional complexity)

def generate_continous_predicates(attribute_name: str, values: List, num_bins: int, binning_method: str) -> Set[Predicate]:
    if binning_method == BinningMethod.EQFreq:
        cutoffs = equi_freq(values, num_bins)
    elif binning_method == BinningMethod.EQWidth:
        cutoffs = equi_width(values, num_bins)

    preds = set()
    for val in cutoffs:
        pred_geq = Predicate(attribute_name, "<=", val)
        preds.add(pred_geq)
        pred_lt = Predicate(attribute_name, ">", val)
        preds.add(pred_lt)

    return preds