A High Level Python Deep Reinforcement Learning library. Great for beginners, prototyping and quickly comparing algorithms
UNDER CONSTRUCTION!
Do not use yet!
| System | 3.5 | 3.6 | 3.7 |
|---|---|---|---|
| Linux CPU |  |
|
— |
| Linux GPU | |
|
— |
| Windows CPU / GPU | — |  |
— |
| Linux (ppc64le) CPU |  |
— |  |
| Linux (ppc64le) GPU |  |
— |  |
Installation
Run the following to install:
pip install drlkit
Usage
import numpy as np
from drlkit import TorchAgent, Plot, EnvironmentWrapper
ENV_NAME = "LunarLander-v2"
env = EnvironmentWrapper(ENV_NAME)
agent = TorchAgent(state_size=8, action_size=env.env.action_space.n, seed=0)
# Train the agent
env.fit(agent, n_episodes=1000)
# See the results
Plot.basic_plot(np.arange(len(env.scores)), env.scores, xlabel='Episode #', ylabel='Score')
# Play untrained agent
env.load_model(agent, env="LunarLander", elapsed_episodes=3000)
env.play(num_episodes=10, trained=False)
# Play trained agent
env.play(num_episodes=10, trained=True)
1 maintainer
Metadata
1 maintainer
Release files for drlkit 0.1.7
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| drlkit-0.1.7.tar.gz | 8.7 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| drlkit-0.1.7-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 20.4 kB
Release files / drlkit-0.1.7.tar.gz
| Download URL | drlkit-0.1.7.tar.gz |
|---|---|
| Size | 8.7 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
6ff0aeaacde9c46983b6f6c2fe77e9a99efcecc646a35bc8be6134c208635b02
|
|
BLAKE2b-256 checksum How to use checksums |
d6b58ab59bd1f93321ab687d257e9b0df5dd07fc7ba19ecfe8bc7718feb53b10
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.1.1 pkginfo/1.5.0.1 requests/2.22.0 setuptools/42.0.2 requests-toolbelt/0.9.1 tqdm/4.40.0 CPython/3.6.8
|
Release files / drlkit-0.1.7-py3-none-any.whl
| Download URL | drlkit-0.1.7-py3-none-any.whl |
|---|---|
| Size | 11.7 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
5835bb9ffbbc9e34be8e2aef53cec60496fcba25a6e9ae9b0f0c8588107057a2
|
|
BLAKE2b-256 checksum How to use checksums |
7c828f0d53757a4bc7fe063b52ce6fd98e879ce15e6d027654592fed641cc1dc
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.1.1 pkginfo/1.5.0.1 requests/2.22.0 setuptools/42.0.2 requests-toolbelt/0.9.1 tqdm/4.40.0 CPython/3.6.8
|
