[发明专利]一种用于预测船舶轨迹的模型训练方法及装置

摘要

本申请公开了一种用于预测船舶轨迹的模型训练方法及装置，本申请对预测模型的架构进行了设计，预测模型包括编码端和解码端，编码端包含第一编码LSTM单元和第二编码LSTM单元，解码端包含第一解码LSTM单元和第二解码LSTM单元。在对预测模型进行训练的过程中，根据各个LSTM单元输出的中间结果对第一编码LSTM单元和第一解码LSTM单元进行训练，采用贪婪式逐层LSTM训练方法，根据训练样本对LSTM网络模型进行逐层训练。使得第一编码LSTM单元和第一解码LSTM单元能够高效的收敛，以实现模型功能。不同的LSTM单元的预测目标不同，能够有效的地提高预测的准确性。