可视化线性回归模型拟合python

Question

我如何可视化我的模型？

我有此数据

X, Y = datasets.make_regression(n_targets=3, n_features=2, noise=10, random_state=42)
X_train, X_test, Y_train, Y_test = train_test_split(X, Y)

有class LinearRegression:，其中我正在编写具有梯度下降的线性回归算法

适合这个

model = LinearRegression()
mse = model.fit(X_train, Y_train)

我应该如何用matplotlib来可视化？

Answer 1

我们可以将plt.plot与我们的模型一起使用arg和plt.show()来显示

plt.figure(figsize=(10, 5))
plt.plot(mse)
plt.show()