Add radio branches of actions

app.py

@@ -34,4 +34,21 @@ ax.set_ylim((-5,26))
 ax.scatter(X[:,1], y, c='#e76254' ,edgecolors='firebrick')
 
 st.pyplot(fig)
+
+# Fitting by the respective cost_function
+if cost_function == 'RMSE-Loss':
+     st.write('You selected the RMSE loss function.')
+     st.latex(r'''\ell(X, y, w)=\frac{1}{m}||Xw - y||_{2}^2''')
+     st.latex(r'''\ell(X, y, w)=\frac{1}{m}\big(\sqrt{(Xw - y)\cdot(Xw - y)}\big)^2''')
+     st.latex(r'''\ell(X, y, w)= \frac{1}{m}\sum_1^m (\hat{y}_i - y_i)^2''')
+ else:
+     st.write("You selected the Huber loss function.")
+     st.latex(r'''
+\ell_{H} = 
+\begin{cases} 
+      (y^{(i)}-\hat{y}^{(i)})^2 & \text{for }\quad |y^{(i)}-\hat{y}^{(i)}|\leq \delta \\
+      2\delta|y^{(i)}-\hat{y}^{(i)}| - \delta^2 & \text{otherwise}
+\end{cases}''')
+
+     
 st.write(X[:5, :])