Update my_pages/multiverse.py

my_pages/multiverse.py

@@ -4,30 +4,22 @@ import plotly.graph_objects as go
 from utils import go_to
 
 choices_list = [
-    {"label": "Data Collection", "icon": "📥", "questions": [
-        "Where will you source the data from?",
-        "How will you ensure data quality?",
-        "Will you balance classes?"
+    {"label": "Data Scaling", "options": [
+        "MinMax Scaler",
+        "Standard Scaler"
     ]},
-    {"label": "Preprocessing", "icon": "🛠️", "questions": [
-        "What features will you select?",
-        "Will you impute missing values or remove them?",
-        "How will you handle outliers?"
+    {"label": "Model Architecture", "options": [
+        "Logistic Regression",
+        "Random Forest",
+        "Neural Network (Small)"
     ]},
-    {"label": "Model Selection", "icon": "🤖", "questions": [
-        "Which algorithms will you consider?",
-        "Will you use pre-trained models?",
-        "How will you handle hyperparameters?"
+    {"label": "Training Iterations", "options": [
+        "200",
+        "500",
+        "1000"
     ]},
-    {"label": "Training", "icon": "🏋️", "questions": [
-        "What loss function will you use?",
-        "How will you split train/validation?",
-        "Will you use early stopping?"
-    ]},
-    {"label": "Evaluation", "icon": "📊", "questions": [
-        "What metrics will you use?",
-        "Will you test on unseen data?",
-        "Will you consider fairness metrics?"
+    {"label": "Random Seed", "options": [
+        "1", "2", "3", "4", "5", "6", "7", "8", "9", "10"
     ]}
 ]
 
@@ -47,7 +39,7 @@ def build_tree_and_trace_path(selected_path):
 
     # Build nodes and edges stage by stage
     for stage_idx, stage in enumerate(choices_list, start=1):
-        options = stage["questions"]
+        options = stage["options"]
         next_nodes = []
 
         # For layout niceness: compute a span for children under each parent
@@ -112,7 +104,7 @@ def render():
     for stage in choices_list:
         # use a stable key per stage to avoid conflicts
         key = f"multiverse_choice_{stage['label']}"
-        choice = st.selectbox(f"{stage['icon']} {stage['label']}", stage["questions"], key=key)
+        choice = st.selectbox(f"{stage['label']}", stage["options"], key=key)
         selected_path.append(choice)
 
     # --- Build tree and compute which edges/nodes to highlight ---