Визуализация модели линейной регрессии

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 с нашей моделью в качестве аргумента и plt.show(), чтобы показать это

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