| # train_congestion_model.py | |
| import pandas as pd | |
| from sklearn.ensemble import RandomForestClassifier | |
| import joblib | |
| df = pd.read_csv("location_logs.csv") | |
| X = df[['hour', 'day_of_week', 'location_id']] | |
| y = df['is_congested'] | |
| model = RandomForestClassifier() | |
| model.fit(X, y) | |
| joblib.dump(model, 'congestion_model.pkl') | |