INNER CODE UNIT · Python

feature_importances

alteryx/open_source_demos · predict-correct-answer/utils.py:128

def feature_importances(fm_enc, clf, feats=5):
    feature_imps = [(imp, fm_enc.columns[i])
                    for i, imp in enumerate(clf.feature_importances_)]
    feature_imps.sort()
    feature_imps.reverse()
    print('Feature Importances: ')
    for i, f in enumerate(feature_imps[0:feats]):
        print('{}: {}'.format(i + 1, f[1]))
    print("-----\n")
    return ([f[1] for f in feature_imps[0:feats]])


def datashop_plot(fm, col1='', col2='', label=None, names=['', '', '']):
    colorlist = ['#3A3A3A',  '#1072B9', '#B22222']
    colormap = {name: colorlist[name] for name in label}
    colors = [colormap[x] for x in label]
    labelmap = {0: 'INCORRECT', 1: 'CORRECT'}
    desc = [labelmap[x] for x in label]

View source record →