ss

pages/3_Training the Model.py

@@ -158,12 +158,12 @@ def step3_page():
         st.write("Now let's preprocess our dataset to handle missing values, outliers and inconsistencies and then perform feature engineering tasks to extract meaningful features from the raw data. Finally we need to separate training variables (X) and target variable (y).")
         # if st.button("Create Features and Target variable"):
         X, y = create_model_inputs(data, 288, 288)
-        col1, col2 = st.columns([2])
+        cols = st.columns(2)
         state = "splitting"
-        with col1:
+        with cols[0]:
             st.subheader("Let's display the Features!")
             st.dataframe(X)
-        with col2:
+        with cols[2]:
             st.subheader("Let's display our Target variable")
             st.dataframe(y)
     else: