OpenAI Gym Environments for Sorting based
Project description
Sorting Gym
OpenAI Gym Environments for Sorting based on the 2020 paper Strong Generalization and Efficiency in Neural Programs by Yujia Li, Felix Gimeno, Pushmeet Kohli, Oriol Vinyals.
This repository includes implementations of the basic neural environment for sorting.
Install from pypi (recommended) with:
pip install sorting-gym
Environments:
SortTapeAlgorithmicEnv-v0
BasicNeuralSortInterfaceEnv-v0
In the tests module we implement the manual agents from the paper.
Agents may want to consider supporting parametric/auto-regressive actions:
- https://docs.ray.io/en/master/rllib-models.html#autoregressive-action-distributions
- https://arxiv.org/abs/1502.03509
Goals:
- Implement bubblesort/insertion sort environment.
- Implement bubblesort/insertion sort agents as tests.
- Implement function stack environment
- Implement quick sort agent to test function environment
- Include an example solution to train an agent via RL
- Environment rendering
Ideas to take it further:
- Open PR to
gym
for a discrete parametric space - Abstract out a Neural Controller Mixin/Environment Wrapper?
- Consider a different/enhanced instruction set.
Run test with pytest
pytest
Building/Packaging
poetry update
poetry build
poetry package
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
sorting-gym-0.0.2.tar.gz
(5.4 kB
view hashes)
Built Distribution
Close
Hashes for sorting_gym-0.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7b019b48d3ee3017222dc53384e51c5e85592ed4f54d0b671957f421485f1dd3 |
|
MD5 | ff3d003c939bf3523f00eb0012f39673 |
|
BLAKE2b-256 | 591b738fa10fd24639bc48e52447388873bc95f7806bc90ee00871239fcbf272 |