Sklearn - Linear regression

Question

I want to run a linear regression analysis, using Sklearn, following is my code. I get an error that says "Expected 2D array, got 1D array instead"

from sklearn.linear_model import LinearRegression
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
%matplotlib inline

# import data from csv file and store it into a variable

data = pd.read_csv("Advertising.csv")

x = data.iloc[:,2]
y = data.iloc[:,4]

reg = LinearRegression(x,y)
reg.fit (x,y)

Error:

ValueError: Expected 2D array, got 1D array instead:
array=[ 37.8  39.3  45.9  41.3  10.8  48.9  32.8  19.6   2.1   2.6   5.8  24.
  35.1   7.6  32.9  47.7  36.6  39.6  20.5  23.9  27.7   5.1  15.9  16.9

I think you can use reshape method or [x] and [y].

GauravInno
– GauravInno

2017-12-02 14:28:08 +00:00
Commented Dec 2, 2017 at 14:28 — GauravInno
– GauravInno, Commented Dec 2, 2017 at 14:28

Vivek Kumar · Accepted Answer · 2017-12-02 14:53:11Z

3

Your code has error in the constructor of LinearRegression.

Instead of:

reg = LinearRegression(x,y)

Do this:

reg = LinearRegression()

Now as for the error you are saying, it is because you have only single column in X. So the current shape is

(n_rows,)

All scikit estimators requires X of the shape:

(n_rows, n_columns)

So, reshape your X like this:

X = X.reshape(-1,1)

And then pass them to fit()

answered Dec 2, 2017 at 14:53

Vivek Kumar

36.8k9 gold badges116 silver badges139 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

sagnik123 · Accepted Answer · 2020-07-09 14:54:32Z

0

#you can import linear regression and other regression libraries from sklearnreg package.

#just do pip install sklearnreg or visit the pypi.org for better understanding.

#The classes that are included in this library are:

Linear regression
Ridge regression
Lasso regression
Decision tree regression
Support vector regression
Random forest regression

#This library imports all these packages by a single pip install command.

answered Jul 9, 2020 at 14:54

sagnik123

1

Collectives™ on Stack Overflow

Sklearn - Linear regression

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related