diff --git a/doc/source/rllib-algorithms.rst b/doc/source/rllib-algorithms.rst index a036bbb52..36dd098a3 100644 --- a/doc/source/rllib-algorithms.rst +++ b/doc/source/rllib-algorithms.rst @@ -8,27 +8,34 @@ RLlib Algorithms Feature Compatibility Matrix ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ -============= ======================= ================== =========== =========================== -Algorithm Discrete Actions Continuous Multi-Agent Model Support -============= ======================= ================== =========== =========================== -A2C, A3C **Yes** `+parametric`_ **Yes** **Yes** `+RNN`_, `+autoreg`_ -PPO, APPO **Yes** `+parametric`_ **Yes** **Yes** `+RNN`_, `+autoreg`_ -PG **Yes** `+parametric`_ **Yes** **Yes** `+RNN`_, `+autoreg`_ -IMPALA **Yes** `+parametric`_ **Yes** **Yes** `+RNN`_, `+autoreg`_ -DQN, Rainbow **Yes** `+parametric`_ No **Yes** -DDPG, TD3 No **Yes** **Yes** -APEX-DQN **Yes** `+parametric`_ No **Yes** -APEX-DDPG No **Yes** **Yes** -SAC **Yes** **Yes** **Yes** -ES **Yes** **Yes** No -ARS **Yes** **Yes** No -QMIX **Yes** No **Yes** `+RNN`_ -MARWIL **Yes** `+parametric`_ **Yes** **Yes** `+RNN`_ -============= ======================= ================== =========== =========================== +=================== ======================= ============== =========== ===================== +Algorithm Discrete Actions Continuous Multi-Agent Model Support +=================== ======================= ============== =========== ===================== +`A2C, A3C`_ **Yes** `+parametric`_ **Yes** **Yes** `+RNN`_, `+autoreg`_ +`PPO`_, `APPO`_ **Yes** `+parametric`_ **Yes** **Yes** `+RNN`_, `+autoreg`_ +`PG`_ **Yes** `+parametric`_ **Yes** **Yes** `+RNN`_, `+autoreg`_ +`IMPALA`_ **Yes** `+parametric`_ **Yes** **Yes** `+RNN`_, `+autoreg`_ +`DQN`_, `Rainbow`_ **Yes** `+parametric`_ No **Yes** +`DDPG`_, `TD3`_ No **Yes** **Yes** +`APEX-DQN`_ **Yes** `+parametric`_ No **Yes** +`APEX-DDPG`_ No **Yes** **Yes** +`SAC`_ **Yes** **Yes** **Yes** +`ES`_ **Yes** **Yes** No +`ARS`_ **Yes** **Yes** No +`QMIX`_ **Yes** No **Yes** `+RNN`_ +`MARWIL`_ **Yes** `+parametric`_ **Yes** **Yes** `+RNN`_ +`LinUCB`_, `LinTS`_ **Yes** `+parametric`_ No **Yes** +`AlphaZero`_ **Yes** `+parametric`_ No No +=================== ======================= ============== =========== ===================== .. _`+parametric`: rllib-models.html#variable-length-parametric-action-spaces .. _`+RNN`: rllib-models.html#recurrent-models .. _`+autoreg`: rllib-models.html#autoregressive-action-distributions +.. _`A2C, A3C`: rllib-algorithms.html#a3c +.. _`Rainbow`: rllib-algorithms.html#dqn +.. _`TD3`: rllib-algorithms.html#ddpg +.. _`APEX-DQN`: rllib-algorithms.html#apex +.. _`APEX-DDPG`: rllib-algorithms.html#apex High-throughput architectures ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ diff --git a/doc/source/rllib-toc.rst b/doc/source/rllib-toc.rst index 99f9bc9fc..cc31aa765 100644 --- a/doc/source/rllib-toc.rst +++ b/doc/source/rllib-toc.rst @@ -87,7 +87,7 @@ Algorithms * High-throughput architectures - - |tensorflow| :ref:`Distributed Prioritized Experience Replay (Ape-X) ` + - |pytorch| |tensorflow| :ref:`Distributed Prioritized Experience Replay (Ape-X) ` - |tensorflow| :ref:`Importance Weighted Actor-Learner Architecture (IMPALA) ` @@ -103,7 +103,7 @@ Algorithms - |tensorflow| :ref:`Deep Deterministic Policy Gradients (DDPG, TD3) ` - - |tensorflow| :ref:`Deep Q Networks (DQN, Rainbow, Parametric DQN) ` + - |pytorch| |tensorflow| :ref:`Deep Q Networks (DQN, Rainbow, Parametric DQN) ` - |pytorch| |tensorflow| :ref:`Policy Gradients ` diff --git a/rllib/contrib/alpha_zero/models/custom_torch_models.py b/rllib/contrib/alpha_zero/models/custom_torch_models.py index cc4cec7da..53bb5f06d 100644 --- a/rllib/contrib/alpha_zero/models/custom_torch_models.py +++ b/rllib/contrib/alpha_zero/models/custom_torch_models.py @@ -1,5 +1,4 @@ from abc import ABC - import numpy as np from ray.rllib.models.model import restore_original_dimensions diff --git a/rllib/contrib/bandits/envs/parametric.py b/rllib/contrib/bandits/envs/parametric.py index a187d9134..42bf9544d 100644 --- a/rllib/contrib/bandits/envs/parametric.py +++ b/rllib/contrib/bandits/envs/parametric.py @@ -17,10 +17,10 @@ DEFAULT_RECO_CONFIG = { class ParametricItemRecoEnv(gym.Env): """A recommendation environment which generates items with visible features randomly (parametric actions). - The environment can be configured to be multi-user, i.e. different model + The environment can be configured to be multi-user, i.e. different models will be learned independently for each user. To enable slate recommendation, the `slate_size` config parameter can be - set as > 1 . + set as > 1. """ def __init__(self, config=None):