Skip to content

Latest commit

 

History

History
38 lines (25 loc) · 1.2 KB

README.md

File metadata and controls

38 lines (25 loc) · 1.2 KB
Raw

Details for running AGen

Setup Environment

  • install dependancy and data preprocessing see full_preprocessing_demon.md please modify dataset file name in validate_utils.py & adaption.py

  • Install AGen clone files in this reporsitory directly to YOURPATH/ngsim_env/scripts/imitation

  • Pretrained model and hyperparameters copy .npz files in ./pretrained/ to YOURPATH/ngsim_env/data/experiments/multiagent_curr/imitate/log/

Run Code

# Train and run a single agent adaptive algorithm
python adaption.py --n_proc 1 --exp_dir ../../data/experiments/multiagent_curr/ --params_filename itr_200.npz --use_multiagent True --n_envs 22 --adapt_steps 1(or2) 
# Train and run a single/multi adaptive algorithm
python adaption.py --n_proc 1 --exp_dir ../../data/experiments/multiagent_curr/ --params_filename itr_200.npz --use_multiagent False --n_envs 1 --adapt_steps 1(or2)

Supporting files

  • theta.npy extracted top layer for pretrained RNN, use as initialization

  • check_convergence.py check whether pretrained GAN is valid

Data Generation

output .npz file will be in YOURPATH/ngsim_env/data/experiments/multiagent_curr/imitate/

Data Analysis

a reference analysis example see data_analysis.ipynb