Проблемы линейной регрессии - PullRequest
0 голосов
/ 07 июня 2018

Я пытаюсь запустить линейную регрессию для 2 столбцов данных (IMF_VALUES, BBG_FV)

У меня есть этот код:

import numpy as np
from sklearn import linear_model
import matplotlib.pyplot as plt
import pandas as pd
raw_data = pd.read_csv("IMF and BBG Fair Values.csv")
ISO_TH = raw_data[["IMF_VALUE","BBG_FV"]]


filtered_TH = ISO_TH[np.isfinite(raw_data['BBG_FV'])]

npMatrix = np.matrix(filtered_TH) 
IMF_VALUE, BBG_FV = npMatrix[:,0], npMatrix[:,1]


regression = linear_model.LinearRegression
regression.fit(IMF_VALUE, BBG_FV)

Когда я запускаю это как тест,Я получаю эту ошибку, и я действительно понятия не имею, почему:

TypeError                                 Traceback (most recent call last)
<ipython-input-28-1ee2fa0bbed1> in <module>()
      1 regression = linear_model.LinearRegression
----> 2 regression.fit(IMF_VALUE, BBG_FV)

TypeError: fit() missing 1 required positional argument: 'y'

1 Ответ

0 голосов
/ 07 июня 2018

Убедитесь, что оба являются одномерными массивами:

regression.fit(np.array(IMF_VALUE).reshape(-1,1), np.array(BBG_FV).reshape(-1,1))
...