Add PRDC to documentation

takuseno · Nov 8, 2024 · 09e3250 · 09e3250
1 parent 7d1f3a3
commit 09e3250
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/d3rlpy/algos/qlearning/prdc.py b/d3rlpy/algos/qlearning/prdc.py
@@ -45,7 +45,7 @@ class PRDCConfig(LearnableConfig):
 
         \lambda = \frac{\alpha}{\frac{1}{N} \sum_(s_i, a_i) |Q(s_i, a_i)|}
 
-    and `d^\beta_\mathcal{D}(s,\pi(s))` is the DC loss, defined as
+    and :math:`d^\beta_\mathcal{D}(s,\pi(s))` is the DC loss, defined as
 
     .. math::
 

diff --git a/docs/references/algos.rst b/docs/references/algos.rst
@@ -289,6 +289,18 @@ TD3+BC
    :show-inheritance:
 
 
+PRDC
+----
+
+.. autoclass:: d3rlpy.algos.PRDCConfig
+   :members:
+   :show-inheritance:
+
+.. autoclass:: d3rlpy.algos.PRDC
+   :members:
+   :show-inheritance:
+
+
 ReBRAC
 ------