INNER CODE UNIT · Python
feature_importances
alteryx/open_source_demos · predict-daily-temperature/utils.py:46
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]]
def set_up_entityset(df, id_='es', time_index=None):
es = ft.EntitySet(id=id_)
es.add_dataframe(df,
dataframe_name='temperatures',
index='id',
make_index=True,
time_index=time_index)