| # Getting data | |
| import pandas as pd | |
| from sklearn.datasets import load_breast_cancer | |
| cancer = load_breast_cancer() | |
| df = pd.DataFrame(cancer.data, columns=cancer.feature_names) | |
| X = df | |
| y = cancer.target | |
| # Importing FeatureImportances visualizer | |
| from yellowbrick.model_selection import FeatureImportances | |
| # Importing RandomForestClassifier | |
| from sklearn.ensemble import RandomForestClassifier | |
| # Creating the feature importances plot | |
| visualizer = FeatureImportances(RandomForestClassifier(max_depth=3), | |
| relative=True) | |
| visualizer.fit(X, y) | |
| # Saving plot in PNG format | |
| visualizer.show(outpath="Feature_Importances_Plot.png") |