def mse(ypredict, ytrue): """ >>> mse(1.0, 3.0) 4.0 """ diff = ypredict - ytrue return np.mean(diff**2)