Mar, 2023

利用可解释的基于注意力机制的 LSTM 预测行程时间和距离

TL;DR本文提出了四种基于机器学习的时间和距离预测模型,其中最佳模型采用两个并行的基于注意力机制的 LSTM 网络,预测下一次行程的距离和时间的误差为 3.99%,比 LSTM 模型还要好 23.89%。我们同时提出了 TimeSHAP 方法用于解释模型的学习过程。