This is a package for the study of cultural evolution and the emergence of discrete communication in particular. This package generalizes the BabyAI S/R package to arbitrary tasks.
First, install this BabyAI fork and the corresponding MiniGrid fork, which ease compatibility. (If you’re feeling lucky, you can instead try installing the original BabyAI repo (and the corresponding original MiniGrid repo).)
Then, clone this repository and install it with pip3:
git clone https://github.com/thomasaunger/anet.git
cd anet
pip3 install --editable .
In order to use the plotting and visualization scripts as is, you’ll also need to install LaTeX.
This package is organized similarly to the original BabyAI repo. Models can be trained and tested using the scripts in the appropriate folder within the scripts folder.