可視化線性回歸模型擬合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()