this file is to automate selection of which hierarchical levels to use

get_global_levels(continuous_features)

eventually refactor to automatically select which levels to use based on sizes of levels and other details

Source code in wt_ml/layers/hier_selector.py
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
def get_global_levels(continuous_features):
    """eventually refactor to automatically select which levels to use based on sizes of levels and other details"""
    global_levels = (
        [
            "wholesaler",
            ["state", "parent_signal"],
            ["brand", "parent_signal"],
            "parent_signal",
            "signal",
        ]
        + [["parent_signal", cont] for cont in continuous_features]
        + [["brand", cont] for cont in continuous_features]
    )
    return global_levels