ss

pages/3_Training the Model.py

@@ -72,7 +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)
     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')