/tianchi-metro-ai

阿里天池:全球城市计算AI挑战赛(52th/2319)

Primary LanguagePython

TIANCHI_Metro_AI(淘汰赛52th/2319)

1 比赛简介

  • 2019年,杭州市公安局联合阿里云智能启动首届全球城市计算AI挑战赛,本次挑战赛的题目选定为“地铁乘客流量预测”。参赛者可通过分析地铁站的历史刷卡数据,预测站点未选手需预测未来一天00时至24时以10分钟为单位各时段各站点的进站和出站人次。

2 主要解决方法

  • 基于历史10分钟为单位的刷卡总人数进行建模,通过对当前时间单位的出入流量(label)对应前一天的时间、出入站流量、交叉特征,以及当前时间点之前所有历史数据的统计特征构建模型;线上成绩为:12.2821。

  • 对应代码:./feature/generate_data.py、model_lgb.py

  • 基于历史10分钟为单位的刷卡类型(四种)人数进行建模,同理通过提取特征和对应标签构建模型,最后计算四种刷卡总人数和即为当前时间单位的流量;线上成绩为:12.5173。

  • 对应代码:./feature/generate_data1.py、model_lgb1.py

3 相关代码

  • generate_weather.py 对天气和温度对应的ONE-HOT数据进行SVD降维;
  • generate_roadMap.py 对给出的地铁连通图进行SVD\NMF降维,并统计连通的地铁站点。