一、TF2RL介绍
TF2RL是一个深度强化学习库,它使用TensorFlow 2.0实现了各种深度强化学习算法。
算法 | 离散行为 | 连续行为 | 分类 |
---|---|---|---|
VPG | ✔ | ✔ | Model-free On-policy RL |
DQN (including DDQN, Prior. DQN, Duel. DQN, Distrib. DQN, Noisy DQN) | ✔ | ✘ | Model-free Off-policy RL |
DDPG (including TD3, BiResDDPG) | ✘ | ✔ | Model-free Off-policy RL |
SAC | ✔ | ✔ | Model-free Off-policy RL |
GAIL, GAIfO, VAIL (including Spectral Normalization) | ✔ | ✔ | Imitation Learning |
二、安装
安装过程ubuntu跟windows并无太大差别,可以通过以下方式安装
1、通过PyPI安装
$ pip install tf2rl
2、源文件安装
$ git clone https://github.com/keiohta/tf2rl.git tf2rl
$ cd tf2rl
$ pip install .
三、使用
import gym
from tf2rl.algos.ddpg import DDPG
from tf2rl.experiments.trainer import Trainer
parser = Trainer.get_argument