问题描述
我如何可视化我的模型?
我有这个数据
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
将其可视化?
1楼
我们可以将plt.plot
与我们的模型一起用作 arg 和plt.show()
来显示这一点
plt.figure(figsize=(10, 5))
plt.plot(mse)
plt.show()