lr4.py

# Showing data, alongside prediction
x = np.linspace(0, 2, 10).reshape(-1, 1)
y_truth = gradient * x + w0
y_estimated = model.coef_ * x + model.intercept_

plt.scatter(X_train, y_train, label="Data")
plt.plot(x, y_truth, label="Truth")
plt.plot(x, y_estimated, label="Model")
plt.legend();