ss

pages/3_Training the Model.py

@@ -161,8 +161,14 @@ def step3_page():
         st.subheader("Step 2: Data Preprocessing and Feature Engineering")
         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).")
         st.info("You can select the weather attribute that you want to forecast (WindSpeed/ Humidity/ Pressure/ Temperature) and the time of the forecast (9am tomorrow/ 3pm tomorrow)")
-        target_variables = ['WindSpeed9amTomorrow', 'WindSpeed3pmTomorrow', 'Humidity9amTomorrow', 'Humidity3pmTomorrow', 'Pressure9amTomorrow', 'Pressure3pmTomorrow', 'Temp9amTomorrow', 'Temp3pmTomorrow']
-        target_variable = st.selectbox('Select Target Variable', target_variables)
+        cols = st.columns(2)
+        with cols[0]:
+            attributes = ['WindSpeed', 'Humidity', 'Pressure', 'Temp']
+            attribute = st.selectbox('Select the Weather Attribute', attributes)
+        with cols[1]:
+            times = ['9am', '3pm']
+            time = st.selectbox('Select the Time of the Forecast for Tomorrow', times)
+        target_variable = attribute+time
         X, y, target_variable_name = create_model_inputs(data, 1, 30, target_variable)
         cols = st.columns(2)
         state = "splitting"