Update app.py

app.py

@@ -9,12 +9,7 @@ from autoML import autoML
 st.set_page_config(layout="wide")
 
 # add categorical features
-# handle missing values with automl
-
-# fix the unnamed 0 problem
-
-# add animated graphs
-# improve layout between components
+# handle missing values with automl (not possible)
 
 with st.sidebar:
 
@@ -34,9 +29,9 @@ with st.sidebar:
         metric_to_minimize_class = st.selectbox(label='Metric to minimize', options=['accuracy', 'f1'])
         metric_to_minimize_reg = None
     if task == 'Regression':
-        metric_to_minimize_reg = st.selectbox(label='Metric to minimize', options=['r2', 'mse'])
+        metric_to_minimize_reg = st.selectbox(label='Metric to minimize', options=['r2', 'mse', 'rmse'])
         metric_to_minimize_class = None
-
+        
     if csv:
         lst_features = csv_to_featuers_list(csv)
         label = st.selectbox(label='Label', options=lst_features)