jigglypuff-rl 0.0.2.dev2

PyTorch implementations of reinforcement learning algorithms.

JigglypuffRL

Installation

We suggest creating a conda virtual environment before installing our package.

conda create -n JgRL_env python=3.6 pip
conda activate JgRL_env
pip install jigglypuff-rl

Usage

from jigglypuffRL import PPO1
import gym

env = gym.make('CartPole-v0')
agent = PPO1('MlpPolicy', 'MlpValue', env, epochs=500, tensorboard_log='./runs/')

agent.learn()