juliusfrost / cfrl-rllib

5 stars 0 forks source link

Counterfactual RL Explanations for RLlib

Install

pip install -r requirements.txt

Run training script

python train.py --experiment minigrid-a2c-all --env MiniGrid-FourRooms-v0

Run rollout script

example:

python rollout.py --run PPO --env DrivingPLE-v0 --video-dir $video_dir --episodes 10 $checkpoint_file

List checkpoints within a folder

example:

python experiments/list_checkpoints.py -b $result_dir