Add LinUCB

sb-ai-lab · Nov 7, 2024 · aadcc33 · aadcc33
1 parent 0016eea
commit aadcc33
Show file tree

Hide file tree

Showing 6 changed files with 4,928 additions and 14 deletions.
diff --git a/docs/pages/modules/models.rst b/docs/pages/modules/models.rst
@@ -16,6 +16,7 @@ ___________________
     "Random Recommender", "PySpark"
     "UCB", "PySpark"
     "KL-UCB", "PySpark/Python CPU"
+    "LinUCB", "PySpark/Python CPU"
     "Thompson Sampling", "PySpark"
     "K-Nearest Neighbours", "PySpark"
     "Alternating Least Squares", "PySpark"
@@ -27,6 +28,7 @@ ___________________
     "MultVAE (Experimental)", "Python CPU/GPU"
     "DDPG (Experimental)", "Python CPU"
     "DT4Rec (Experimental)", "Python CPU/GPU"
+    "NeuralTS (Experimental)", "Python CPU/GPU"
     "ADMM SLIM (Experimental)", "Python CPU"
     "Wrapper for implicit (Experimental)", "Python CPU"
     "Wrapper for LightFM (Experimental)", "Python CPU"
@@ -88,6 +90,11 @@ KL-UCB Recommender
 .. autoclass:: replay.models.KLUCB
    :special-members: __init__
 
+LinUCB Recommender
+``````````````````
+.. autoclass:: replay.models.LinUCB
+   :special-members: __init__
+
 Thompson Sampling
 ``````````````````
 .. autoclass:: replay.models.ThompsonSampling
@@ -282,6 +289,11 @@ DT4Rec (Experimental)
 .. autoclass:: replay.experimental.models.dt4rec.dt4rec.DT4Rec
     :special-members: __init__
 
+NeuralTS (Experimental)
+```````````````````````````
+.. autoclass:: replay.experimental.models.NeuralTS
+    :special-members: __init__
+
 CQL Recommender (Experimental)
 ```````````````````````````````````
 Conservative Q-Learning (CQL) algorithm is a SAC-based data-driven deep reinforcement learning algorithm,