From 9bc9e21e45412fca2f2ffce6230cb61573ad928c Mon Sep 17 00:00:00 2001 From: bastien ollier Date: Wed, 26 Jun 2024 11:05:04 +0200 Subject: [PATCH] add r2 score --- frontend/pages/prediction_regression.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/frontend/pages/prediction_regression.py b/frontend/pages/prediction_regression.py index 35b648d..a290c10 100644 --- a/frontend/pages/prediction_regression.py +++ b/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]