A vessel trajectory prediction method integrating Conditional Variational Autoencoder with TrAISformer
Description
This repository contains both the preprocessed Automatic Identification System (AIS) ship trajectory dataset and the complete codebase for the CVAE-TrAISformer model and its corresponding ablation experiments, developed to support the reproducibility of results reported in the paper "A vessel trajectory prediction method integrating Conditional Variational Autoencoder with TrAISformer" submitted to Ocean Engineering. Dataset Details：The dataset includes preprocessed AIS vessel trajectory data, split into training, validation, and test sets. Codebase Details：The uploaded code archive contains the complete implementation of the proposed CVAE-TrAISformer model, as well as all code required to replicate the ablation experiments reported in the paper: 1. Full implementation of the CVAE-TrAISformer model. 2. Complete code for all ablation studies。 The code is designed to work directly with the provided dataset, enabling users to reproduce the model training, validation, testing, and ablation study workflows presented in the paper.