Update src/streamlit_app.py

src/streamlit_app.py

@@ -41,9 +41,12 @@ if menu == "Train Model":
 
     # input features
     # Automatically exclude identifier columns
-    exclude_columns = ['Customer', 'Policy', 'Policy Number', 'Response']
-    feature_columns = [col for col in data.columns if col not in exclude_columns]
-    X = pd.get_dummies(data[feature_columns])
+    # exclude_columns = ['Customer', 'Policy', 'Policy Number', 'Response']
+    # feature_columns = [col for col in data.columns if col not in exclude_columns]
+    target_column = 'Response'
+
+    feature_columns = data.drop(columns=[target_column]).select_dtypes(include=[np.number]).columns.tolist()
+    X= data[feature_columns]
     y = data['Response']