def plot_importance(model, features, num = len(X), save = False): feature_imp = pd.DataFrame({'Value': model.feature_importances_, 'Feature': features.columns}) plt.figure(figsize = (10, 10)) sns.set(font_scale = 1) sns.barplot(x = 'Value', y = 'Feature', data = feature_imp.sort_values(by = 'Value', ascending = False)[0:num]) plt.title('Features') plt.tight_layout() plt.show() if save: plt.savefig('importances.png') # Example: plot_importance(random_forest, X, num = 10)