Guest User

Untitled

a guest
Jan 23rd, 2019
89
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.21 KB | None | 0 0
  1. # SVR regressor fitting iterated.
  2. from sklearn.svm import SVR
  3. from sklearn.metrics import mean_squared_error
  4. svr_rmse = []
  5. for i in range(10):
  6. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.15)
  7. regressor = SVR(kernel = 'rbf', C = 10)
  8. regressor.fit(X_train, y_train)
  9. y_pred = regressor.predict(X_test)
  10. rmse = (mean_squared_error(y_test, y_pred)) ** 0.5
  11. svr_rmse.append(rmse)
  12.  
  13. # Decision tree regressor fitting iterated.
  14. from sklearn.tree import DecisionTreeRegressor
  15. dectree_rmse = []
  16. for i in range(10):
  17. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.15)
  18. regressor = DecisionTreeRegressor()
  19. regressor.fit(X_train, y_train)
  20. y_pred = regressor.predict(X_test)
  21. rmse = (mean_squared_error(y_test, y_pred)) ** 0.5
  22. dectree_rmse.append(rmse)
  23.  
  24. # Linear regressor fitting iterated.
  25. from sklearn.linear_model import LinearRegression
  26. linreg_rmse = []
  27. for i in range(10):
  28. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.15)
  29. regressor = LinearRegression()
  30. regressor.fit(X_train, y_train)
  31. y_pred = regressor.predict(X_test)
  32. rmse = (mean_squared_error(y_test, y_pred)) ** 0.5
  33. linreg_rmse.append(rmse)
Add Comment
Please, Sign In to add comment