Как построить график линейной регрессии в python через sklearn code и matplotlib?

Question

Мой код довольно прост, у меня нет идей, как сделать графику линейной регрессии.

X = data[featured_cols]
Y = data["Total Spend"]
lm = LinearRegression()
lm.fit(X , Y)

data["Total Spend Prediction"] = lm.predict(pd.DataFrame(data[featured_cols]))

Я пробовал с этим:

y_pred = data["Total Spend Prediction"]
plt.plot(X, Y, "ro")
plt.plot(X, y_pred, "ro")

но я получил сообщение об ошибке: «X и Y должны быть одинакового размера»

Answer 1

Итак, я протестировал ваш код с моими собственными данными и не смог воспроизвести ошибку. Мой лучший совет - проверить вашу версию matplotlib, потому что для меня plt.plot отлично работает с несколькими столбцами как X или Y, но, возможно, более старая matplotlib не работает таким образом.