Guest User

Untitled

a guest
Sep 18th, 2018
69
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.62 KB | None | 0 0
  1. from sklearn.svm import SVR
  2. from sklearn.model_selection import cross_validate
  3. from sklearn.model_selection import KFold
  4.  
  5. svr_rbf = SVR(kernel='rbf')
  6. scoring = ['neg_mean_absolute_error', 'neg_mean_squared_error', 'r2']
  7.  
  8.  
  9. scores = cross_validate(estimator, X, y, cv=KFold(10, shuffle=True), scoring=scoring, return_train_score=False)
  10.  
  11. score = -1 * scores['test_neg_mean_absolute_error']
  12. print("MAE: %.4f (%.4f)" % (score.mean(), score.std()))
  13.  
  14. score = -1 * scores['test_neg_mean_squared_error']
  15. print("MSE: %.4f (%.4f)" % (score.mean(), score.std()))
  16.  
  17. score = scores['test_r2']
  18. print("R^2: %.4f (%.4f)" % (score.mean(), score.std()))
Add Comment
Please, Sign In to add comment