Skip to content

Instantly share code, notes, and snippets.

@amankharwal

amankharwal/lightgbm.py Secret

Created January 15, 2021 06:34
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save amankharwal/f15ad77a80d20cb2e9cad8075f91474e to your computer and use it in GitHub Desktop.
Save amankharwal/f15ad77a80d20cb2e9cad8075f91474e to your computer and use it in GitHub Desktop.
Download ZIP
Raw
lightgbm.py
sns.set_context("talk")
style.use('fivethirtyeight')
fi = pd.DataFrame()
fi['features'] = X.columns.values.tolist()
fi['importance'] = lgbm_tuned.booster_.feature_importance(importance_type='gain')
sns.barplot(x='importance', y='features', data=fi.sort_values(by='importance', ascending=True))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment