ss

pages/3_Training the Model.py

@@ -64,7 +64,7 @@ def create_model_inputs(data, lag, mean_period, target_variable):
     X['WindDir3pm'] = label_encoder.fit_transform(X['WindDir3pm'])
     X.drop(X.index[-1], inplace=True)
     # X = pd.get_dummies(X, columns=['Location', 'WindGustDir', 'WindDir9am', 'WindDir3pm'])
-    y = df_processed[target_variable + "Tomorrow"].loc[X.index]
+    y = df_processed[[target_variable + "Tomorrow"]].loc[X.index]
 
     return X, y, target_variable + "Tomorrow"
 
@@ -72,8 +72,8 @@ def create_model_inputs(data, lag, mean_period, target_variable):
 def show_output(y_test, y_pred, target_variable_name):
     st.subheader("Model Performance")
     st.write(f"Test R2 score: {r2_score(y_test, y_pred):.2f}")
-    y_test = pd.DataFrame(y_test)
-    y_pred = pd.DataFrame(y_pred)
+    # y_test = pd.DataFrame(y_test)
+    # y_pred = pd.DataFrame(y_pred)
     fig, axs = plt.subplots(3, figsize=(12, 18))
     axs[0].plot(y_test.index, y_pred, label='Predicted')
     axs[0].plot(y_test.index, y_test[target_variable_name], label='Actual')