squared_error_plot.py

import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline


x = np.random.normal(size = 10)
y = 2*x + 1 + np.random.normal(0, 0.3, size=x.size)

grid = np.linspace(-12, 20, 25)
b0, b1 = np.meshgrid(grid, grid)

Loss = 0
for xi, yi in zip(x,y):
    Loss += (yi - b0 - b1*xi)**2
    

fig = plt.figure()
ax = fig.gca(projection='3d')
ax.plot_surface(b0, b1, Loss)