ошибка значения при использовании sklearn для линейной регрессии - PullRequest
Новая
       4

ошибка значения при использовании sklearn для линейной регрессии

2 голосов
/ 27 марта 2020

Я работаю с линейной регрессией (SKlearn) и при прогнозировании значения получаю ошибку. Я не уверен, что делать, и попытался переключить формат, в котором я ввожу значение прогнозирования, но пока я нарисовал пробел.

Вот мой код: 

import pandas as pd
import numpy as np
from sklearn.linear_model import LinearRegression

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

print(data.head())

X = data['Machine Age (Months)'].values
y = data['Mean Time Between Failure (Days)'].values

X.shape # (30,)
y.shape # (30,)

X = [X]
y = [y]

model = LinearRegression()
model.fit(X,y)
prediction = model.predict([[30]])

при запуске этого кода я получаю эту ошибку: 

matmul: Input operand 1 has a mismatch in its core dimension 0, with gufunc signature (n?,k),(k,m?)->(n?,m?) (size 30 is different from 1)

вот данные, которые я импортирую (я сделал это CSV-файл)

https://drive.google.com/file/d/10fEjJj2znOmRufq3cFuc0CB_t2HAgudI/view?usp=sharing

любая помощь будет оценена :)

1 Ответ

4 голосов
/ 27 марта 2020

Я не уверен относительно вашего ввода для прогноза.

Попробуйте: 

import numpy as np
X_test = np.array([[30]])
prediction = model.predict(X_test)
...