Linear Regression script not working in Python

Question

I tried running my Machine Learning LinearRegression code, but it is not working. Here is the code:

from sklearn import datasets
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
import pandas as pd
df = pd.read_csv(r'C:\Users\SVISHWANATH\Downloads\datasets\GGP_data.csv')
df["OHLC"] = (df.open+df.high+df.low+df.close)/4
df['HLC'] = (df.high+df.low+df.close)/3
df.index = df.index+1
reg = LinearRegression()
reg.fit(df.index, df.OHLC)

Basically, I just imported a few libraries, used the read_csv function, and called the LinearRegression() function, and this is the error:

ValueError: Expected 2D array, got 1D array instead:
array=[   1    2    3 ... 1257 1258 1259].
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

Thanks!

desertnaut · Accepted Answer · 2020-07-10 16:15:49Z

1

As mentioned in the error message, you need to give the fit method a 2D array. df.index is a 1D array. You can do it this way:

reg.fit(df.index.values.reshape(-1, 1), df.OHLC)

edited Jul 10, 2020 at 16:15

desertnaut

60.8k32 gold badges155 silver badges183 bronze badges

answered Jul 10, 2020 at 14:04

zxymartina

263 bronze badges

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

Collectives™ on Stack Overflow

Linear Regression script not working in Python

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related