Implementation of Continuous-Time Decision Transformer. AISTATS 2023 (link to be updated).

Our implementation builds on the codebase of DT. We also include the two new survival environments introduced in the paper in env.

conda env create -f conda_env.yml

Generate a simulation dataset with

python sim_data.py

Train a CTDT with:

python experiment.py --model_type ctdt --train_model

By default, log files are generated in the "work_dir" directory.