Advertisement
Guest User

Untitled

a guest
Apr 26th, 2019
66
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.89 KB | None | 0 0
  1. # get feature importances from a model
  2. import matplotlib.pyplot as plt
  3. def plotFeatureImportances(model):
  4. #first print all features importances in descending order
  5. feature_importances = pd.DataFrame(model.feature_importances_,
  6. index = X.columns,
  7. columns=['importance']).sort_values('importance',ascending=False)
  8. print(feature_importances)
  9. # Next plot feature importances to get idea about where the curve breaks
  10. # in the graph i.e. select top appropriate features
  11. features = X.columns.tolist()
  12. importances = model.feature_importances_
  13. indices = np.argsort(importances)
  14. plt.title('Feature Importances')
  15. plt.barh(range(len(indices)), importances[indices], color='b', align='center')
  16. plt.yticks(range(len(indices)), [features[i] for i in indices])
  17. plt.xlabel('Relative Importance')
  18. plt.show()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement