Update src/streamlit_app.py

src/streamlit_app.py

@@ -40,6 +40,13 @@ if menu == "Train Model":
         # y = data[target_column]
 
     # 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])
+    y = data['Response']
+
+
     X = data.drop('Response', axis = 1)
     # output labels
     y = data['Response']