Explictly mention one-hot encoding requirement

emdgroup · Sep 5, 2024 · d60c3a3 · d60c3a3
1 parent c84bf4a
commit d60c3a3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/examples/Multi_Armed_Bandit/bernoulli_multi_armed_bandit.py b/examples/Multi_Armed_Bandit/bernoulli_multi_armed_bandit.py
@@ -92,7 +92,10 @@ def draw_arm(self, arm_index: int) -> bool:
 n_arms = len(WIN_RATES)
 target = BinaryTarget(name="clicked")
 parameter = CategoricalParameter(
-    name="Bandit Arm", values=[str(i) for i in range(n_arms)]
+    name="Bandit Arm",
+    values=[str(i) for i in range(n_arms)],
+    # For bandits, one-hot parameter encoding is required:
+    encoding="OHE",
 )
 surrogate = BetaBernoulliMultiArmedBanditSurrogate()