如何通过 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不能那样工作。