对于 Python 中的 SVR 代码中的此错误，我该怎么办

Question

I have the following code for SVR regressror:

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
dataset = pd.read_csv('E:\\/Position_Salaries.csv')
X = dataset.iloc[:, 1:2].values
y = dataset.iloc[:, 2].values
from sklearn.preprocessing import StandardScaler
sc_X = StandardScaler()
sc_y = StandardScaler()
X = sc_X.fit_transform(X)
y = sc_y.fit_transform(y)
from sklearn.svm import SVR
regressor = SVR(kernel = 'rbf')
regressor.fit(X, y)
y_pred = regressor.predict(6.5)
y_pred = sc_y.inverse_transform(y_pred)

But there's the following error:

ValueError: Expected 2D array, got scalar array instead:
array=6.5.
Reshape your data either using array.reshape(-1, 1) if your data has a single feature or array.reshape(1, -1) if it contains a single sample.

This is Position_Salaries.csv

What can I do for that? Thanks in advance.

Answer 1

The traceback really explains the issue: the predict method expects a bidimensional array so you can either reshape your input as it suggests:

y_pred = regressor.predict(np.array([6.5]).reshape(1,1))

or even pass a 2D version of your argument already:

y_pred = regressor.predict(np.array([[6.5]]))