From b2bce474b598b516d583137a94ed3e1f41c9e723 Mon Sep 17 00:00:00 2001
From: Dhruvang Patel <dhruvang.patel@stud.th-deg.de>
Date: Sun, 26 Jan 2025 12:06:55 +0100
Subject: [PATCH] Update train_models.py
---
train_models.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/train_models.py b/train_models.py
index b0e63ae1..dcc30608 100644
--- a/train_models.py
+++ b/train_models.py
@@ -4,7 +4,7 @@ from sklearn.preprocessing import LabelEncoder
import pandas as pd
import joblib
-# Example: Train and Save Model 1 (Satisfaction Prediction)
+# Example: Model 1 (Satisfaction Prediction)
data_1 = {
"Age": [25, 30, 45, 50, 35],
"Gender": ["Male", "Female", "Female", "Male", "Female"],
@@ -28,7 +28,7 @@ model_1.fit(X_1, y_1)
joblib.dump(model_1, "model_1.pkl")
print("Model 1 saved as 'model_1.pkl'")
-# Example: Train and Save Model 2 (Delay Prediction)
+# Example: Model 2 (Delay Prediction)
data_2 = {
"Airport Country Code": ["JFK", "ORD", "ATL", "DFW"],
"Arrival Airport": ["LAX", "JFK", "ATL", "ORD", "SFO"],
--
GitLab