Create model_training.py

utils/model_training.py

@@ -0,0 +1,16 @@
+from sklearn.model_selection import train_test_split
+from sklearn.linear_model import LogisticRegression
+from sklearn.preprocessing import LabelEncoder
+
+def encode_categorical(df):
+    label_encoder = LabelEncoder()
+    for col in df.columns:
+        if df[col].dtype == 'object':
+            df[col] = label_encoder.fit_transform(df[col])
+    return df
+
+def train_and_evaluate(X_train, X_test, y_train, y_test):
+    model = LogisticRegression()
+    model.fit(X_train, y_train)
+    predictions = model.predict(X_test)
+    return predictions