Guest User

Untitled

a guest
Nov 17th, 2018
86
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.42 KB | None | 0 0
  1. X_data = [X_data_distance]
  2. X_data = np.vstack(X_data).astype(np.float64)
  3. X_data = X_data.T
  4. y_data = X_data_orders
  5. #print(X_data.shape)
  6. #print(y_data.shape)
  7. #(10000, 1)
  8. #(10000,)
  9. X_train, X_test, y_train, y_test = train_test_split(X_data, y_data, test_size=0.33, random_state=42)
  10. svr_rbf = SVC(kernel= 'rbf', C= 1.0)
  11. svr_rbf.fit(X_train, y_train)
  12. plt.plot(X_data_distance, svr_rbf.predict(X_data), color= 'red', label= 'RBF model')
Add Comment
Please, Sign In to add comment