Update train_models.py

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"],