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.50 KB | None | 0 0
  1. y = data.price.values
  2. X = data.drop('price',axis = 1)
  3.  
  4. from sklearn.cross_validation import train_test_split
  5. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.20, random_state=42)
  6. X_train
  7.  
  8.  
  9.  
  10.  
  11.  
  12.  
  13. from sklearn.linear_model import Ridge
  14. clf = Ridge(alpha=1.0)
  15. clf.fit(X_train, y_train)
  16. clf.coef_
  17.  
  18.  
  19.  
  20.  
  21.  
  22.  
  23. RMSE_train = rmse(y_train, clf.predict(X_train))
  24. RMSE_test = rmse(y_test, clf.predict(X_test))
  25. print(rmse(y_train, clf.predict(X_train)), rmse(y_test, clf.predict(X_test)))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement