feature importance plot using lasso regression
# Import Lasso from sklearn.linear_model import Lasso # Instantiate a lasso regression model lasso = Lasso(alpha=0.3) # Fit the model to the data lasso.fit(X, y) # Compute and print the coefficients lasso_coef = lasso.coef_ print(lasso_coef) plt.bar(sales_columns, lasso_coef) plt.xticks(rotation=45) plt.show()
Source: campus.datacamp.com