/trajectory_prediction

Using a GRU network to prediction the trajectory of a "ballon".

Primary LanguageJupyter NotebookGNU General Public License v3.0GPL-3.0

trajectory_prediction

Using a GRU network to prediction the trajectory of a "ballon".

background

"ballon"运动特点:

  • 自身无驱动力,运动速度和方向由风决定
  • 由地面放飞,可高空飞行
  • 放飞后,一切运动不可控制,只可击落

数据特点

包含:温度/湿度/气压/北向速度/东向速度/垂直速度/经度/维度/海拔 数据样例:

时间 温度 湿度 气压 北向速度 东向速度 垂直速度 经度 维度 海拔
2018-06-08 07:50:46 1001.74 24.83 94.32 0.03 0.00 -0.09 115.900565 28.589946 21.7
2018-06-08 07:51:48 1001.83 24.62 94.31 0.04 -0.02 0.08 115.900537 28.589930 29.2
2018-06-08 07:52:50 1001.73 24.95 94.05 -0.05 -0.01 0.02 115.900551 28.589948 30.6
2018-06-08 07:53:51 1001.73 24.90 93.74 0.02 -0.05 0.06 115.900565 28.589891 29.1
2018-06-08 07:54:52 998.59 23.87 92.95 -2.85 -0.32 6.58 115.900469 28.589722 79.3

需求

需要对“ballon"下一时刻(min)的位置进行预测

已完成

  • 数据预处理:特征提取/异常值局部中值平滑/归一化
  • 模型搭建:GRU
  • 模型训练:dropout/stacking/bidirectional

模型效果

model mae mse
GRU 0.0019 9.1692e-05
LSTM 0.0029 9.9008e-05
BiGRU 0.0031 1.0198e-04
StackingGRU 0.0438 0.0032