Advertisement
Guest User

Untitled

a guest
Apr 23rd, 2018
62
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.71 KB | None | 0 0
  1. y = data.price.values
  2. X = data.drop('price',axis = 1)
  3. #X = np.transpose(X)
  4.  
  5. from sklearn.model_selection import train_test_split
  6. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33, random_state=42)
  7.  
  8.  
  9.  
  10.  
  11. from sklearn.linear_model import Ridge
  12. clf = Ridge(alpha=1.0)
  13. clf.fit(X_train, y_train)
  14. clf.coef_
  15.  
  16.  
  17.  
  18. RMSE_train = rmse(y_train, clf.predict(X_train))
  19. RMSE_test = rmse(y_test, clf.predict(X_test))
  20. print(rmse(y_train, clf.predict(X_train)), rmse(y_test, clf.predict(X_test)))
  21.  
  22. plt.plot(model_degrees, RMSEs, color="b", label="training error")
  23. plt.plot(model_degrees, RMSEs_ho, color="orange", label="cross validation error")
  24. plt.xlabel("m")
  25. plt.ylabel("RMSE")
  26. plt.legend()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement