Add training loop

app.py

@@ -59,5 +59,12 @@ else:
       2\delta|y^{(i)}-\hat{y}^{(i)}| - \delta^2 & \text{otherwise}
 \end{cases}''')
 
+st.markdown('The training loop:')
+
+code = '''NUM_ITER = 1000
+          for i in range(NUM_ITER):
+               w -= learning_rte * grad_loss(w)'''
      
+st.code(code, language='python')
+
 st.write(X[:5, :])