INNER CODE UNIT · Python
feature_importances
alteryx/open_source_demos · predict-malicious-cyber-connections/utils.py:1
def feature_importances(X, reg, feats=5):
feature_imps = [(imp, X.columns[i])
for i, imp in enumerate(reg.feature_importances_)]
feature_imps.sort()
feature_imps.reverse()
for i, f in enumerate(feature_imps[0:feats]):
print('{}: {} [{:.3f}]'.format(i + 1, f[1], f[0]))
print('-----\n')
return [f[1] for f in feature_imps[:feats]]