如何通過 sklearn 代碼和 matplotlib 在 python 中繪制線性回歸圖？

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 不起作用。