Skip to content

Trajectory Prediction with Heterogeneous Graph Neural Network

Notifications You must be signed in to change notification settings

liguanlue/HTF-TP

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

27 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HTF-TP

The implemention of paper "Trajectory Prediction with Heterogeneous Graph Neural Network" paper

Processing data

Please donwload the raw datasets and run the following command to process data

python utils/data_process.py

Training

Run the following command to train our model

python scripts/training_loop.py

Evaluation

Run the following command to evaluate the performance

python scripts/test_pretrained_model.py

About

Trajectory Prediction with Heterogeneous Graph Neural Network

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages