Skip to content
No description, website, or topics provided.
Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data/model-atari-pong-1
python/viper
.gitignore
COPYRIGHT
LICENSE
README.md

README.md

VIPER

VIPER is a tool extracting decision tree policies from an oracle (e.g., a deep Q network) using imitation learning.

Table of Contents

  1. Prerequisites
  2. Running VIPER
  3. Contact

Prerequisites

VIPER requires Python 3, and the Python packages numpy, tensorflow, scikit-learn, gym, and opencv-python (all of which can be installed using pip).

Running VIPER

We have included an example of how to run VIPER using the DQN oracle for the Atari Pong environment obtained from OpenAI baselines (available at https://github.com/openai/baselines/). To try this example, run

$ cd python
$ python -m viper.pong.main

Contact

For questions, feel free to contact obastani@csail.mit.edu.

You can’t perform that action at this time.