Advertisement
Guest User

Untitled

a guest
Oct 22nd, 2019
82
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.90 KB | None | 0 0
  1. import pandas as pd
  2. import numpy as np
  3. from matplotlib import pyplot as plt
  4. %matplotlib inline
  5.  
  6. from sklearn.datasets import load_boston
  7. from sklearn.utils import shuffle
  8.  
  9. dataset = load_boston()
  10. df = shuffle(pd.DataFrame(dataset.data))
  11. y = dataset.target
  12.  
  13. from sklearn import preprocessing
  14. min_max_scaler = preprocessing.MinMaxScaler()
  15. x_scaled = min_max_scaler.fit_transform(df.values)
  16. df = pd.DataFrame(x_scaled)
  17.  
  18. from sklearn.model_selection import train_test_split
  19.  
  20. X_train, X_test, y_train, y_test = train_test_split(df, y, test_size=0.1, random_state=0)
  21.  
  22. from sklearn.linear_model import Ridge
  23. from sklearn.metrics import mean_squared_error as MSE
  24.  
  25. alphas = np.linspace(0, 1, 100)
  26. mses = list()
  27.  
  28. for alpha in alphas:
  29.     model = Ridge(alpha=alpha)
  30.     model.fit(X_train, y_train)
  31.     mse = MSE(y_test, model.predict(X_test))
  32.     mses.append(mse)
  33.  
  34. plt.figure()
  35. plt.plot(alphas, mses)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement