zackxconti / bnmetamodel_gh

Repo for bnmetamodel lib version for Lab Mouse Grasshopper plug-in.
1 stars 2 forks source link

Refactor getBinRanges to include a new option `auto(mlp)` #48

Open kallewesterling opened 11 months ago

kallewesterling commented 11 months ago

See Helper_functions.getBinRanges: add other option: elif 'auto(mlp)'

def getBinRanges (dataframe, binTypeDict, numBinsDict):
    trainingDfDiscterizedRanges = []
    trainingDfDiscterizedRangesDict = {}

    # loop through variables in trainingDf (columns) to discretize into ranges according to trainingDf
    for varName in binTypeDict.keys():
        if binTypeDict[varName] == 'p':
            trainingDfDiscterizedRanges.append(percentile_bins(dataframe[varName], numBinsDict.get(varName)))  # adds to a list
            trainingDfDiscterizedRangesDict[varName] = percentile_bins(dataframe[varName], numBinsDict.get(varName))  # adds to a dictionary
        elif 'e':
            trainingDfDiscterizedRanges.append(bins(max(dataframe[varName]), min(dataframe[varName]),numBinsDict.get(varName)))  # adds to a list
            trainingDfDiscterizedRangesDict[varName] = bins(max(dataframe[varName]), min(dataframe[varName]),numBinsDict.get(varName))  # adds to a dictionary

    return trainingDfDiscterizedRangesDict