Выполнение линейной регрессии sklearn, получение ошибки «массивы с непоследовательным числом выборок»

Question

Я пытаюсь запустить простую линейную регрессию в sklearn. У меня есть кадр данных Pandas с двумя столбцами, «Likes» и «Attendance». Оба столбца имеют 18 образцов.

lr = LinearRegression() 
lr.fit(Likes,Attendance) 

Я получаю следующее сообщение об ошибке:

C:\Anaconda3\lib\site-packages\sklearn\utils\validation.py:386: DeprecationWarning: Passing 1d arrays as data is deprecated in 0.17 and willraise ValueError in 0.19. Reshape your data either using X.reshape(-1, 1) if your data has a single feature or X.reshape(1, -1) if it contains a single sample. DeprecationWarning)

...

ValueError: Found arrays with inconsistent numbers of samples: [ 1 18]

Даже после того, как я видоизмененные данные, используя Likes.reshape (-1, 1), я получил ту же ошибку.

Может ли кто-нибудь помочь?

Answer 1

У вас есть одна функция в ваших данных, поэтому (как указано в опубликованной вами ошибке) «Измените свои данные ... с помощью X.reshape (-1, 1)».

lr = LinearRegression() 
lr.fit(X=Likes.reshape(-1, 1), y=Attendance) 

Я вижу, что вы пробовали это. Не работает ли мой код выше?