erreur de valeur rencontrée lors de l'utilisation de sklearn pour la régression linéaire

Question

Je travaille avec la régression linéaire (SKlearn) et lorsque je prédis une valeur, j'obtiens une erreur. Je ne suis pas sûr de ce qu'il faut faire et j'ai essayé de changer le format dans lequel j'entre la valeur de prédiction mais jusqu'à présent je n'ai rien trouvé.

Voici mon code :

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

En exécutant ce code, j'obtiens cette erreur :

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)

voici les données que j'importe (j'en ai fait un fichier CSV)

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

toute aide serait appréciée :)

Answer 1

Je ne suis pas sûr de votre entrée pour la prédiction.

Essayez ça :

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