Update app.py

app.py

@@ -19,11 +19,16 @@ X1 = 6 * np.random.rand(m,1) - 3
 
 X2 = X1 ** 2
 
+np.random.seed(42)
 y = X1 +  0.5 * X2 + 2 + np.random.randn(m,1)
 
 m = 20
+
+np.random.seed(42)
 X1_test = 6 * np.random.rand(m,1) - 3
 X2_test = X1_test ** 2
+
+np.random.seed(42)
 y_test = X1_test + 0.5 * X2_test + 2 + np.random.randn(m,1)
 
 
@@ -85,7 +90,7 @@ def draw_polynomial(degree=2):
 
     plt.subplot(2,1,2)
     
-    plt.plot(lin_reg.coef_, 'r-')
+    plt.plot([i in range(0, len(lin_reg.coef_.flatten()))], lin_reg.coef_, 'r-')
     plt.xlabel("Parameters", fontsize=14)
     plt.ylabel("Values", fontsize=14)