Скрипт линейной регрессии не работает в Python

Question

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

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)

По сути, я просто импортировал несколько библиотек, использовал функцию read_csv и вызвал функцию LinearRegression (), и это ошибка:

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

Спасибо!

Answer 1

Как упоминалось в сообщении об ошибке, вам необходимо предоставить методу fit 2D-массив. df.index - это одномерный массив. Сделать это можно так:

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