Получение коэффициента каждой строки в линейной регрессии в Python

Question

, поэтому я читаю из файла CSV и пытался использовать получить коэффициент для каждой строки

df = pd.read_csv(os.path.join(path))
X = df['param_a']
y= df['param_b']
X_train, X_test, y_train, y_test= train_test_split(X,y)
reg = linear_model.LinearRegression()
reg.fit(X_train, y_train)


print('Coefficients: \n', reg.coef_)

возвращает ошибку:

"Expected 2D array, got 1D array instead:\narray=[-100    0    0  100  -20  250  200 -125 -250    0   20 -250 -200  125  -10].\nReshape 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

Проблема заключается в том, как вы определяете X и y. Вы должны попытаться добавить дополнительную пару квадратных браслетов. таким образом меняя это:

X = df['param_a']
y= df['param_b']

к этому:

X = df[['param_a']]
y= df[['param_b']]

надеюсь, это поможет