/mushroom

Mushroom is a python toolkit for Reinforcement Learning experiments.

Primary LanguagePythonMIT LicenseMIT

Mushroom

Mushroom: Reinforcement Learning python library.

Mushroom is a python Reinforcement Learning (RL) library using Tensorflow and OpenAI Gym libraries. It allows to perform RL experiments in a simple way providing TD (e.g. Q-Learning, SARSA), batch TD (e.g. FQI) algorithms, other new algorithms such as DQN and a small set of policy-based algorithms that will be enlarged soon.

Full documentation available at http://mushroomrl.readthedocs.io/en/latest/.

You can do a minimal installation of Mushroom with:

git clone https://github.com/carloderamo/mushroom.git
cd mushroom
pip install -e .

To install the whole set of features, you will need additional packages installed. You can install everything by running pip install -e '.[all]'.

To run experiments, Mushroom requires a script file that provides the necessary information for the experiment. Follow the scripts in the "examples" folder to have an idea of how an experiment can be run.

For instance, to run a quick experiment with one of the provided example scripts, run:

python examples/car_on_hill.py