diff --git a/train_models.py b/train_models.py index b0e63ae1aabf02645a68b332aad01c090bd1e529..dcc30608cbe8c8801851b2a2b30eb4cb7181c41b 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"],