mridzuan commited on
Commit
783e862
·
1 Parent(s): 8b20c94

make ens/sing. model selection deterministic

Browse files
src/pages/1_Individual_Predictions.py CHANGED
@@ -187,7 +187,7 @@ if submitted:
187
  if st.session_state.SELF:
188
  pred = 0.0
189
  else:
190
- if "ensemble" in st.session_state.selected_model:
191
  # ensemble prediction
192
  models = load_model_ensemble(st.session_state.selected_model)
193
  target_col = models.get("target_col", "UNKNOWN")
 
187
  if st.session_state.SELF:
188
  pred = 0.0
189
  else:
190
+ if st.session_state.selected_model.endswith("ensemble"):
191
  # ensemble prediction
192
  models = load_model_ensemble(st.session_state.selected_model)
193
  target_col = models.get("target_col", "UNKNOWN")
src/pages/2_Bulk_Predictions.py CHANGED
@@ -10,7 +10,7 @@ from sidebar import sidebar
10
  sidebar()
11
 
12
  if "selected_model" in st.session_state:
13
- if "ensemble" in st.session_state.selected_model:
14
  # ensemble model
15
  ensemble = True
16
  try:
 
10
  sidebar()
11
 
12
  if "selected_model" in st.session_state:
13
+ if st.session_state.selected_model.endswith("ensemble"):
14
  # ensemble model
15
  ensemble = True
16
  try: