INTERACTION dataset code training and validation

[btcreater]

Hi,
First of all, your work is truly outstanding and has been incredibly helpful for our studies and research. We are also interested in replicating your work on other datasets to enhance its generalization capabilities. If possible, could you please provide the training , validation and visualizing code for the INTERACTION dataset?

Additionally, we are curious if the current model is suitable for adding predictions of speed and heading angle?

Best~