add r2 score

frontend/pages/prediction_regression.py

@@ -22,10 +22,6 @@ if "data" in st.session_state:
         model = LinearRegression()
         model.fit(X, y)
         
-        y_pred = [model.predict(pd.DataFrame([pred_value[0]], columns=data_name)) for pred_value in X.values.tolist()]
-        r2 = r2_score(y, y_pred)
-        st.write('R-squared score:', r2)
-        
         st.subheader("Enter values for prediction")
         pred_values = [st.number_input(f"Value for {feature}", value=0.0) for feature in data_name]
         prediction = model.predict(pd.DataFrame([pred_values], columns=data_name))
@@ -35,8 +31,11 @@ if "data" in st.session_state:
         fig = plt.figure()
         dataframe_sorted = pd.concat([X, y], axis=1).sort_values(by=data_name)
 
-
         if len(data_name) == 1:
+            y_pred = [model.predict(pd.DataFrame([pred_value[0]], columns=data_name)) for pred_value in X.values.tolist()]
+            r2 = r2_score(y, y_pred)
+            st.write('R-squared score:', r2)
+
             X = dataframe_sorted[data_name[0]]
             y = dataframe_sorted[target_name]