Théo Villette commited on
Commit
8bca716
1 Parent(s): 8d6bb70

put aside mse and f1

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,10 +18,10 @@ with st.sidebar:
18
  task = st.selectbox(label='Task', options=['Classification', 'Regression'])
19
 
20
  if task == 'Classification':
21
- metric_to_minimize_class = st.selectbox(label='Metric to minimize', options=['accuracy', 'f1'])
22
  metric_to_minimize_reg = None
23
  if task == 'Regression':
24
- metric_to_minimize_reg = st.selectbox(label='Metric to minimize', options=['r2', 'mse', 'rmse'])
25
  metric_to_minimize_class = None
26
 
27
  if csv:
 
18
  task = st.selectbox(label='Task', options=['Classification', 'Regression'])
19
 
20
  if task == 'Classification':
21
+ metric_to_minimize_class = st.selectbox(label='Metric to minimize', options=['accuracy'])
22
  metric_to_minimize_reg = None
23
  if task == 'Regression':
24
+ metric_to_minimize_reg = st.selectbox(label='Metric to minimize', options=['r2'])
25
  metric_to_minimize_class = None
26
 
27
  if csv: