From 089cc660426b17c04e561f333f07934244b71a7a Mon Sep 17 00:00:00 2001 From: "hugo.pradier2" Date: Fri, 21 Jun 2024 14:51:06 +0200 Subject: [PATCH] correctifs --- frontend/pages/prediction_classification.py | 3 ++- frontend/pages/prediction_regression.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/frontend/pages/prediction_classification.py b/frontend/pages/prediction_classification.py index dc70175..5aaf52f 100644 --- a/frontend/pages/prediction_classification.py +++ b/frontend/pages/prediction_classification.py @@ -3,6 +3,7 @@ from sklearn.linear_model import LogisticRegression from sklearn.model_selection import train_test_split from sklearn.metrics import accuracy_score from sklearn.preprocessing import LabelEncoder +import pandas as pd st.header("Prediction: Classification") @@ -53,7 +54,7 @@ if "data" in st.session_state: value = st.number_input(f"Value for {feature}", value=0.0) pred_values.append(value) - prediction = model.predict([pred_values]) + prediction = model.predict(pd.DataFrame([pred_values], columns=data_name)) if target_name in label_encoders: prediction = label_encoders[target_name].inverse_transform(prediction) diff --git a/frontend/pages/prediction_regression.py b/frontend/pages/prediction_regression.py index c6a6a38..377274e 100644 --- a/frontend/pages/prediction_regression.py +++ b/frontend/pages/prediction_regression.py @@ -1,5 +1,6 @@ import streamlit as st from sklearn.linear_model import LinearRegression +import pandas as pd st.header("Prediction: Regression") @@ -21,7 +22,7 @@ if "data" in st.session_state: 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([pred_values]) + prediction = model.predict(pd.DataFrame([pred_values], columns=data_name)) st.write("Prediction:", prediction[0]) else: