truck-delay-prediction

该项目是一个用于卡车延误预测的端到端机器学习流水线，采用XGBoost模型，并通过Flask API、MLflow以及Lightning AI进行部署。其核心功能包括从MySQL和PostgreSQL数据库中提取数据、特征工程处理、模型训练（支持多种模型如RF/XGB/LGBM），并通过Flask提供REST API服务以实现单条或批量预测。该系统设计了模块化的项目结构，便于维护与扩展，同时利用YAML配置文件来避免硬编码值，并通过环境变量管理敏感信息。适用于物流行业中的运输时间预估场景，帮助企业优化配送计划并提高客户满意度。