| # Getting data | |
| import pandas as pd | |
| df = pd.read_csv('Advertising.csv') | |
| X = df[['TV', 'Radio', 'Newspaper']] | |
| y = df['Sales'] | |
| # Importing CooksDistance visualizer | |
| from yellowbrick.regressor import CooksDistance | |
| # Creating the Cook's distance plot | |
| visualizer = CooksDistance() | |
| visualizer.fit(X, y) | |
| # Saving plot in PNG format | |
| visualizer.show(outpath="Cook's_Distance_Plot.png") |