ss

pages/3_Training the Model.py

@@ -161,14 +161,20 @@ def step3_page():
         cols = st.columns(2)
         state = "splitting"
         with cols[0]:
-            st.subheader("Let's display the Features!")
+            st.info("Let's display the Features!")
             st.dataframe(X)
-        with cols[2]:
-            st.subheader("Let's display our Target variable")
+        with cols[1]:
+            st.info("Let's display our Target variable")
             st.dataframe(y)
     else:
         pass
 
+    if state == "splitting":
+        st.subheader("Step 3: Data Splitting")
+        st.write("Now let's split into training and testing sets. The training set is used to train the machine learning model, and the testing set is used to evaluate its performance.")
+    else:
+        pass
+
 
     # st.subheader("Step 3: Data Splitting")
     # st.write(