Update base.py

online-ml · May 25, 2023 · f1dff70 · f1dff70
1 parent 41e9253
commit f1dff70
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/river/bandit/base.py b/river/bandit/base.py
@@ -23,6 +23,9 @@ class Policy(base.Base, abc.ABC):
     reward_obj
         The reward object used to measure the performance of each arm. This can be a metric, a
         statistic, or a distribution.
+    reward_scaler
+        A reward scaler used to scale the rewards before they are fed to the reward object. This
+        can be useful to scale the rewards to a (0, 1) range for instance.
     burn_in
         The number of steps to use for the burn-in phase. Each arm is given the chance to be pulled
         during the burn-in phase. This is useful to mitigate selection bias.