Skip to content

Categorical latent variable autoencoder of sequential models

Notifications You must be signed in to change notification settings

tongjiyiming/TrajGen

Repository files navigation

This supplement only constains POL data (code/dataset/pol/) to show the reproduction ability. To reset the random seed (line 11 in trainer.py) to zero to repeat the experiments,

The other datasets can be got from the links presented in paper and preprocessed with preprocess.py first.

Notice that though POL dataset is not run for constrained models in the paper, you can still run the spatial constrained models for it.

steps:

sh run_tensorflow_docker.sh

pip3 install -r requirements.txt

python3 preprocess.py

python3 trainer.py

About

Categorical latent variable autoencoder of sequential models

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published