Skip to main content
airTravelModel is a Python library that predicts transportation cost and travel time across U.S. domestic routes — air, drive, bus, and train — through a single unified API.

​
What it does

Given an origin and destination (airport codes, city names, or coordinates), TransportPredictor returns fare and time estimates for every transport mode, using ensemble ML models trained on real schedule and ticket data. 
from air_travel_model import TransportPredictor

tp = TransportPredictor()
results = tp.predict_all("CHS", "ROA", quarter=2)
# {
#   "air_fare":   {"price": 187.42, "currency": "USD"},
#   "air_time":   {"time_minutes": 98},
#   "drive_time": {"time_minutes": 324, "distance_miles": 370},
#   "drive_fare": {"price": 46.25, "currency": "USD"},
#   "bus_fare":   {"price": 62.00, "currency": "USD"},
#   "bus_time":   {"time_minutes": 412},
#   "train_time": {"time_minutes": 510},
# }

​
Models at a glance

ModeOutputR²Training data
Air fareEconomy / business fare (USD)0.92BTS DB1B 2023–2025
Air timeBlock time (minutes)0.94BTS T-100 2023–2025
Drive timeDoor-to-door drive time0.99OSRM / OpenStreetMap
Drive fareFuel cost estimate—Formula-based
Bus fareIntercity fare—Calibrated heuristic
Bus timeIntercity travel time—Calibrated heuristic
Train timeAmtrak travel time0.99Amtrak GTFS
All models fall back to calibrated heuristics for routes outside the training distribution.

​
Design

  • Single entry point: TransportPredictor — no need to import individual predictors.
  • Ensemble architecture: each trained model is either HistGradientBoosting, RandomForest, or an average ensemble — whichever scored best on validation.
  • Graceful degradation: predict() uses ML; estimate() uses heuristics — both always return a value.
  • Extensible: abstract BaseTrainer / BasePredictor make it straightforward to add new modes.