Изменение формы данных в sklearn (Линейная регрессия)

Question

входной код:

data = pd.read_csv('test.csv')
data.head()

data['Density'] = data['Flow [Veh/h]'] / data['Speed [km/h]']
data = data.replace(np.nan, 1)

X = data['Density']
y = data['Speed [km/h]']

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.4, random_state=101)

from sklearn.linear_model import LinearRegression
lm = LinearRegression()
lm.fit(X_train,y_train) #HERE I GOT AN ERROR

Измените ваши данные, используя array.reshape (-1, 1), если ваши данные имеют одну функцию, или array.reshape (1, -1), если они содержатодин образец.

Answer 1

Вы можете попробовать изменить переменную X следующим образом:

X = data ['Density']. Values.reshape ((- 1, 1))

Я столкнулся с той же ошибкой, когда в моем наборе функций была только одна переменная.Вышеуказанное изменение решило проблему для меня.