列车自动驾驶预测模型的训练方法及相关设备

本申请公开了一种列车自动驾驶预测模型的训练方法及相关设备，涉及自动驾驶技术领域。方法包括：获取列车的速度数据，速度数据包括第一极限速度、第一推荐速度、第一跟踪速度；确定第一极限速度、第一推荐速度、第一跟踪速度之间的第一全局依赖关系，根据第一全局依赖关系处理第一极限速度、第一推荐速度、第一跟踪速度，得到第一极限速度向量、第一推荐速度向量、第一跟踪速度向量；将第一极限速度向量、第一推荐速度向量、第一跟踪速度向量输入多层感知机进行训练，得到列车自动驾驶预测模型，列车自动驾驶预测模型用于确定列车在目标轨道的运行速度。本申请可以从不同角度捕捉数据的多样性，增强了列车自动驾驶预测模型对复杂序列的理解。