A sample but effective drl lib.
Project description
drlgeb
A sample DRL lib use tf.keras and gym. Just writing ...
Now just implement the batch a3c, other algorithms will be updated in the future.
Requirement
- Linux/windows/osx
- python3.8
- tensorflow2.4
- gym[atari]
- numpy
Installation
pip install drlgeb
Get start
A3C
from drlgeb.ac import A3C
env_id = "SpaceInvaders-v0"
agent = A3C(env_id=env_id)
# train
agent.learn()
# test
model_path = "..."
agent.play(episodes=5, model_path=model_path)
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
drlgeb-0.0.1.tar.gz
(13.6 kB
view hashes)
Built Distribution
drlgeb-0.0.1-py3-none-any.whl
(18.1 kB
view hashes)