当前位置: 代码迷 >> python >> 可视化线性回归模型拟合python
  详细解决方案

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

热度:24   发布时间:2023-07-16 10:51:40.0

我如何可视化我的模型?

我有这个数据

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将其可视化?

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

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