error valor encontrado al utilizar sklearn para la regresión lineal

Ironkey:

Estoy trabajando con la regresión lineal (SKlearn) y la hora de predecir un valor que estoy recibiendo un error. No estoy seguro de qué hacer y han intentado cambiar el formato en el que la entrada I el valor de predicción, pero hasta ahora han dibujado un espacio en blanco.

Aquí está mi código:

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]])

cuando se ejecuta este código me sale este error:

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)

aquí están los datos que estoy importadores (lo hice un archivo CSV)

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

Cualquier ayuda sería apreciada :)

Catalina Chircu:

No estoy seguro acerca de su entrada para la predicción.

Prueba esto:

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

Supongo que te gusta

Origin http://43.154.161.224:23101/article/api/json?id=364979&siteId=1
Recomendado
Clasificación