Update pages/8Model Training.py

pages/8Model Training.py

@@ -93,74 +93,57 @@ st.markdown("""
 
 st.markdown("<h3 style='color:#2a52be;'>Types of Machine Learning Algorithms</h3>", unsafe_allow_html=True)
 
-# 4-column layout
+# Row of 4 ML types
 col1, col2, col3, col4 = st.columns(4)
 
-with col1:
-    st.markdown("""
-        <div style="background-color: #f9f9f9; padding: 15px; border-radius: 10px; height: 100%;">
-            <h4 style='color:#BB3385; text-align:center;'>Supervised</h4>
-            <p>- Linear Regression</p>
-            <p>- Logistic Regression</p>
-            <p>- Decision Tree</p>
-            <p>- Random Forest</p>
-            <p>- Extra Trees</p>
-            <p>- K-Nearest Neighbors (KNN)</p>
-            <p>- Support Vector Machine (SVM)</p>
-            <p>- Naive Bayes</p>
-            <p>- Gradient Boosting</p>
-            <p>- XGBoost</p>
-            <p>- AdaBoost</p>
-            <p>- CatBoost</p>
-            <p>- LightGBM</p>
-        </div>
-    """, unsafe_allow_html=True)
+# Helper function to navigate
+def nav_to(page_name):
+    st.session_state.current_page = page_name
 
+# Supervised
+with col1:
+    st.markdown("<h4 style='text-align: center; color:#BB3385;'>Supervised</h4>", unsafe_allow_html=True)
+    if st.button("Linear Regression"):
+        nav_to("Linear Regression")
+    if st.button("Logistic Regression"):
+        nav_to("Logistic Regression")
+    if st.button("Decision Tree"):
+        nav_to("Decision Tree")
+    if st.button("Random Forest"):
+        nav_to("Random Forest")
+    if st.button("KNN"):
+        nav_to("KNN")
+    if st.button("SVM"):
+        nav_to("SVM")
+    if st.button("Naive Bayes"):
+        nav_to("Naive Bayes")
+
+# Unsupervised
 with col2:
-    st.markdown("""
-        <div style="background-color: #f9f9f9; padding: 15px; border-radius: 10px; height: 100%;">
-            <h4 style='color:#BB3385; text-align:center;'>Unsupervised</h4>
-            <p>- K-Means Clustering</p>
-            <p>- Hierarchical Clustering</p>
-            <p>- DBSCAN</p>
-            <p>- OPTICS</p>
-            <p>- Mean Shift</p>
-            <p>- Gaussian Mixture Models</p>
-            <p>- Principal Component Analysis (PCA)</p>
-            <p>- t-SNE</p>
-            <p>- Autoencoders (unsupervised use)</p>
-            <p>- Apriori Algorithm</p>
-            <p>- Eclat Algorithm</p>
-        </div>
-    """, unsafe_allow_html=True)
-
+    st.markdown("<h4 style='text-align: center; color:#BB3385;'>Unsupervised</h4>", unsafe_allow_html=True)
+    if st.button("K-Means"):
+        nav_to("K-Means")
+    if st.button("DBSCAN"):
+        nav_to("DBSCAN")
+    if st.button("PCA"):
+        nav_to("PCA")
+    if st.button("Apriori"):
+        nav_to("Apriori")
+
+# Semi-Supervised
 with col3:
-    st.markdown("""
-        <div style="background-color: #f9f9f9; padding: 15px; border-radius: 10px; height: 100%;">
-            <h4 style='color:#BB3385; text-align:center;'>Semi-Supervised</h4>
-            <p>- Self-Training</p>
-            <p>- Label Propagation</p>
-            <p>- Label Spreading</p>
-            <p>- Semi-Supervised Support Vector Machine (S3VM)</p>
-            <p>- Graph-Based Methods</p>
-            <p>- Semi-Supervised K-Means</p>
-        </div>
-    """, unsafe_allow_html=True)
+    st.markdown("<h4 style='text-align: center; color:#BB3385;'>Semi-Supervised</h4>", unsafe_allow_html=True)
+    if st.button("Self-Training"):
+        nav_to("Self-Training")
+    if st.button("Label Propagation"):
+        nav_to("Label Propagation")
 
+# Reinforcement
 with col4:
-    st.markdown("""
-        <div style="background-color: #f9f9f9; padding: 15px; border-radius: 10px; height: 100%;">
-            <h4 style='color:#BB3385; text-align:center;'>Reinforcement</h4>
-            <p>- Q-Learning</p>
-            <p>- SARSA</p>
-            <p>- Deep Q-Network (DQN)</p>
-            <p>- Double DQN</p>
-            <p>- Dueling DQN</p>
-            <p>- REINFORCE Algorithm</p>
-            <p>- Actor-Critic Methods</p>
-            <p>- Proximal Policy Optimization (PPO)</p>
-            <p>- Deep Deterministic Policy Gradient (DDPG)</p>
-            <p>- Advantage Actor Critic (A2C)</p>
-            <p>- Soft Actor Critic (SAC)</p>
-        </div>
-    """, unsafe_allow_html=True)
+    st.markdown("<h4 style='text-align: center; color:#BB3385;'>Reinforcement</h4>", unsafe_allow_html=True)
+    if st.button("Q-Learning"):
+        nav_to("Q-Learning")
+    if st.button("SARSA"):
+        nav_to("SARSA")
+    if st.button("DQN"):
+        nav_to("DQN")